From 7cfd0da856019773a588e71d813d16be7b75ffef Mon Sep 17 00:00:00 2001 From: CircleCI Docs Date: Fri, 17 Oct 2025 19:16:27 +0000 Subject: [PATCH] rebase --- docs/build/html/.buildinfo | 2 +- .../html/_static/documentation_options.js | 2 +- docs/build/html/cpp/ops.html | 58 +++++++++++++++--- docs/build/html/dev/custom_metal_kernels.html | 8 +-- docs/build/html/dev/extensions.html | 8 +-- docs/build/html/dev/metal_debugger.html | 8 +-- docs/build/html/dev/mlx_in_cpp.html | 8 +-- .../html/examples/linear_regression.html | 8 +-- docs/build/html/examples/llama-inference.html | 8 +-- docs/build/html/examples/mlp.html | 8 +-- docs/build/html/genindex.html | 18 +++--- docs/build/html/index.html | 8 +-- docs/build/html/install.html | 8 +-- docs/build/html/objects.inv | Bin 28497 -> 28708 bytes .../python/_autosummary/mlx.core.Device.html | 10 +-- .../python/_autosummary/mlx.core.Dtype.html | 8 +-- .../_autosummary/mlx.core.DtypeCategory.html | 8 +-- .../python/_autosummary/mlx.core.abs.html | 10 +-- .../python/_autosummary/mlx.core.add.html | 10 +-- .../python/_autosummary/mlx.core.addmm.html | 14 ++--- .../python/_autosummary/mlx.core.all.html | 14 ++--- .../_autosummary/mlx.core.allclose.html | 16 ++--- .../python/_autosummary/mlx.core.any.html | 14 ++--- .../python/_autosummary/mlx.core.arange.html | 12 ++-- .../python/_autosummary/mlx.core.arccos.html | 10 +-- .../python/_autosummary/mlx.core.arccosh.html | 10 +-- .../python/_autosummary/mlx.core.arcsin.html | 10 +-- .../python/_autosummary/mlx.core.arcsinh.html | 10 +-- .../python/_autosummary/mlx.core.arctan.html | 10 +-- .../python/_autosummary/mlx.core.arctan2.html | 10 +-- .../python/_autosummary/mlx.core.arctanh.html | 10 +-- .../python/_autosummary/mlx.core.argmax.html | 14 ++--- .../python/_autosummary/mlx.core.argmin.html | 14 ++--- .../_autosummary/mlx.core.argpartition.html | 14 ++--- .../python/_autosummary/mlx.core.argsort.html | 12 ++-- .../python/_autosummary/mlx.core.array.T.html | 8 +-- .../_autosummary/mlx.core.array.abs.html | 10 +-- .../_autosummary/mlx.core.array.all.html | 10 +-- .../_autosummary/mlx.core.array.any.html | 10 +-- .../_autosummary/mlx.core.array.argmax.html | 10 +-- .../_autosummary/mlx.core.array.argmin.html | 10 +-- .../_autosummary/mlx.core.array.astype.html | 10 +-- .../_autosummary/mlx.core.array.at.html | 8 +-- .../_autosummary/mlx.core.array.conj.html | 10 +-- .../_autosummary/mlx.core.array.cos.html | 10 +-- .../_autosummary/mlx.core.array.cummax.html | 10 +-- .../_autosummary/mlx.core.array.cummin.html | 10 +-- .../_autosummary/mlx.core.array.cumprod.html | 10 +-- .../_autosummary/mlx.core.array.cumsum.html | 10 +-- .../_autosummary/mlx.core.array.diag.html | 10 +-- .../_autosummary/mlx.core.array.diagonal.html | 10 +-- .../_autosummary/mlx.core.array.dtype.html | 8 +-- .../_autosummary/mlx.core.array.exp.html | 10 +-- .../_autosummary/mlx.core.array.flatten.html | 10 +-- .../python/_autosummary/mlx.core.array.html | 14 ++--- .../_autosummary/mlx.core.array.imag.html | 8 +-- .../_autosummary/mlx.core.array.item.html | 8 +-- .../_autosummary/mlx.core.array.itemsize.html | 8 +-- .../_autosummary/mlx.core.array.log.html | 10 +-- .../_autosummary/mlx.core.array.log10.html | 10 +-- .../_autosummary/mlx.core.array.log1p.html | 10 +-- .../_autosummary/mlx.core.array.log2.html | 10 +-- .../mlx.core.array.logcumsumexp.html | 10 +-- .../mlx.core.array.logsumexp.html | 10 +-- .../_autosummary/mlx.core.array.max.html | 10 +-- .../_autosummary/mlx.core.array.mean.html | 10 +-- .../_autosummary/mlx.core.array.min.html | 10 +-- .../_autosummary/mlx.core.array.moveaxis.html | 10 +-- .../_autosummary/mlx.core.array.nbytes.html | 8 +-- .../_autosummary/mlx.core.array.ndim.html | 8 +-- .../_autosummary/mlx.core.array.prod.html | 10 +-- .../_autosummary/mlx.core.array.real.html | 8 +-- .../mlx.core.array.reciprocal.html | 10 +-- .../_autosummary/mlx.core.array.reshape.html | 12 ++-- .../_autosummary/mlx.core.array.round.html | 10 +-- .../_autosummary/mlx.core.array.rsqrt.html | 10 +-- .../_autosummary/mlx.core.array.shape.html | 10 +-- .../_autosummary/mlx.core.array.sin.html | 10 +-- .../_autosummary/mlx.core.array.size.html | 8 +-- .../_autosummary/mlx.core.array.split.html | 10 +-- .../_autosummary/mlx.core.array.sqrt.html | 10 +-- .../_autosummary/mlx.core.array.square.html | 10 +-- .../_autosummary/mlx.core.array.squeeze.html | 10 +-- .../_autosummary/mlx.core.array.std.html | 10 +-- .../_autosummary/mlx.core.array.sum.html | 10 +-- .../_autosummary/mlx.core.array.swapaxes.html | 10 +-- .../_autosummary/mlx.core.array.tolist.html | 12 ++-- .../mlx.core.array.transpose.html | 10 +-- .../_autosummary/mlx.core.array.var.html | 10 +-- .../_autosummary/mlx.core.array.view.html | 10 +-- .../_autosummary/mlx.core.array_equal.html | 12 ++-- .../_autosummary/mlx.core.as_strided.html | 16 ++--- .../_autosummary/mlx.core.async_eval.html | 10 +-- .../_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 | 10 +-- .../_autosummary/mlx.core.bitwise_invert.html | 10 +-- .../_autosummary/mlx.core.bitwise_or.html | 10 +-- .../_autosummary/mlx.core.bitwise_xor.html | 10 +-- .../mlx.core.block_masked_mm.html | 12 ++-- .../mlx.core.broadcast_arrays.html | 12 ++-- .../_autosummary/mlx.core.broadcast_to.html | 12 ++-- .../python/_autosummary/mlx.core.ceil.html | 10 +-- .../_autosummary/mlx.core.clear_cache.html | 10 +-- .../python/_autosummary/mlx.core.clip.html | 10 +-- .../python/_autosummary/mlx.core.compile.html | 20 +++--- .../_autosummary/mlx.core.concatenate.html | 14 ++--- .../python/_autosummary/mlx.core.conj.html | 10 +-- .../_autosummary/mlx.core.conjugate.html | 10 +-- .../_autosummary/mlx.core.contiguous.html | 12 ++-- .../python/_autosummary/mlx.core.conv1d.html | 18 +++--- .../python/_autosummary/mlx.core.conv2d.html | 18 +++--- .../python/_autosummary/mlx.core.conv3d.html | 18 +++--- .../_autosummary/mlx.core.conv_general.html | 22 +++---- .../mlx.core.conv_transpose1d.html | 20 +++--- .../mlx.core.conv_transpose2d.html | 20 +++--- .../mlx.core.conv_transpose3d.html | 20 +++--- .../_autosummary/mlx.core.convolve.html | 12 ++-- .../python/_autosummary/mlx.core.cos.html | 10 +-- .../python/_autosummary/mlx.core.cosh.html | 10 +-- .../mlx.core.cuda.is_available.html | 10 +-- .../python/_autosummary/mlx.core.cummax.html | 16 ++--- .../python/_autosummary/mlx.core.cummin.html | 16 ++--- .../python/_autosummary/mlx.core.cumprod.html | 16 ++--- .../python/_autosummary/mlx.core.cumsum.html | 16 ++--- .../mlx.core.custom_function.html | 8 +-- .../_autosummary/mlx.core.default_device.html | 8 +-- .../_autosummary/mlx.core.default_stream.html | 8 +-- .../python/_autosummary/mlx.core.degrees.html | 10 +-- .../_autosummary/mlx.core.dequantize.html | 16 ++--- .../python/_autosummary/mlx.core.diag.html | 12 ++-- .../_autosummary/mlx.core.diagonal.html | 16 ++--- .../mlx.core.disable_compile.html | 10 +-- .../mlx.core.distributed.Group.html | 8 +-- .../mlx.core.distributed.all_gather.html | 10 +-- .../mlx.core.distributed.all_sum.html | 10 +-- .../mlx.core.distributed.init.html | 14 ++--- .../mlx.core.distributed.is_available.html | 10 +-- .../mlx.core.distributed.recv.html | 14 ++--- .../mlx.core.distributed.recv_like.html | 12 ++-- .../mlx.core.distributed.send.html | 12 ++-- .../python/_autosummary/mlx.core.divide.html | 10 +-- .../python/_autosummary/mlx.core.divmod.html | 12 ++-- .../python/_autosummary/mlx.core.einsum.html | 12 ++-- .../_autosummary/mlx.core.einsum_path.html | 14 ++--- .../_autosummary/mlx.core.enable_compile.html | 10 +-- .../python/_autosummary/mlx.core.equal.html | 10 +-- .../python/_autosummary/mlx.core.erf.html | 10 +-- .../python/_autosummary/mlx.core.erfinv.html | 10 +-- .../python/_autosummary/mlx.core.eval.html | 12 ++-- .../python/_autosummary/mlx.core.exp.html | 10 +-- .../_autosummary/mlx.core.expand_dims.html | 12 ++-- .../python/_autosummary/mlx.core.expm1.html | 10 +-- .../mlx.core.export_function.html | 17 ++--- .../_autosummary/mlx.core.export_to_dot.html | 14 ++--- .../_autosummary/mlx.core.exporter.html | 14 ++--- .../python/_autosummary/mlx.core.eye.html | 16 ++--- .../mlx.core.fast.cuda_kernel.html | 24 ++++---- .../mlx.core.fast.layer_norm.html | 12 ++-- .../mlx.core.fast.metal_kernel.html | 24 ++++---- .../_autosummary/mlx.core.fast.rms_norm.html | 12 ++-- .../_autosummary/mlx.core.fast.rope.html | 22 +++---- ...ore.fast.scaled_dot_product_attention.html | 14 ++--- .../python/_autosummary/mlx.core.fft.fft.html | 14 ++--- .../_autosummary/mlx.core.fft.fft2.html | 14 ++--- .../_autosummary/mlx.core.fft.fftn.html | 14 ++--- .../_autosummary/mlx.core.fft.fftshift.html | 12 ++-- .../_autosummary/mlx.core.fft.ifft.html | 14 ++--- .../_autosummary/mlx.core.fft.ifft2.html | 14 ++--- .../_autosummary/mlx.core.fft.ifftn.html | 14 ++--- .../_autosummary/mlx.core.fft.ifftshift.html | 12 ++-- .../_autosummary/mlx.core.fft.irfft.html | 14 ++--- .../_autosummary/mlx.core.fft.irfft2.html | 14 ++--- .../_autosummary/mlx.core.fft.irfftn.html | 14 ++--- .../_autosummary/mlx.core.fft.rfft.html | 14 ++--- .../_autosummary/mlx.core.fft.rfft2.html | 14 ++--- .../_autosummary/mlx.core.fft.rfftn.html | 14 ++--- .../python/_autosummary/mlx.core.finfo.html | 10 +-- .../python/_autosummary/mlx.core.flatten.html | 14 ++--- .../python/_autosummary/mlx.core.floor.html | 10 +-- .../_autosummary/mlx.core.floor_divide.html | 10 +-- .../python/_autosummary/mlx.core.full.html | 14 ++--- .../_autosummary/mlx.core.gather_mm.html | 12 ++-- .../_autosummary/mlx.core.gather_qmm.html | 20 +++--- .../mlx.core.get_active_memory.html | 10 +-- .../mlx.core.get_cache_memory.html | 10 +-- .../mlx.core.get_peak_memory.html | 10 +-- .../python/_autosummary/mlx.core.grad.html | 14 ++--- .../python/_autosummary/mlx.core.greater.html | 10 +-- .../_autosummary/mlx.core.greater_equal.html | 10 +-- .../mlx.core.hadamard_transform.html | 12 ++-- .../_autosummary/mlx.core.identity.html | 12 ++-- .../python/_autosummary/mlx.core.imag.html | 10 +-- .../mlx.core.import_function.html | 12 ++-- .../python/_autosummary/mlx.core.inner.html | 10 +-- .../python/_autosummary/mlx.core.isclose.html | 16 ++--- .../_autosummary/mlx.core.isfinite.html | 10 +-- .../python/_autosummary/mlx.core.isinf.html | 10 +-- .../python/_autosummary/mlx.core.isnan.html | 10 +-- .../_autosummary/mlx.core.isneginf.html | 10 +-- .../_autosummary/mlx.core.isposinf.html | 10 +-- .../_autosummary/mlx.core.issubdtype.html | 12 ++-- .../python/_autosummary/mlx.core.jvp.html | 16 ++--- .../python/_autosummary/mlx.core.kron.html | 10 +-- .../_autosummary/mlx.core.left_shift.html | 10 +-- .../python/_autosummary/mlx.core.less.html | 10 +-- .../_autosummary/mlx.core.less_equal.html | 10 +-- .../mlx.core.linalg.cholesky.html | 12 ++-- .../mlx.core.linalg.cholesky_inv.html | 12 ++-- .../_autosummary/mlx.core.linalg.cross.html | 12 ++-- .../_autosummary/mlx.core.linalg.eig.html | 10 +-- .../_autosummary/mlx.core.linalg.eigh.html | 12 ++-- .../_autosummary/mlx.core.linalg.eigvals.html | 10 +-- .../mlx.core.linalg.eigvalsh.html | 12 ++-- .../_autosummary/mlx.core.linalg.inv.html | 10 +-- .../_autosummary/mlx.core.linalg.lu.html | 12 ++-- .../mlx.core.linalg.lu_factor.html | 12 ++-- .../_autosummary/mlx.core.linalg.norm.html | 16 ++--- .../_autosummary/mlx.core.linalg.pinv.html | 10 +-- .../_autosummary/mlx.core.linalg.qr.html | 12 ++-- .../_autosummary/mlx.core.linalg.solve.html | 10 +-- .../mlx.core.linalg.solve_triangular.html | 12 ++-- .../_autosummary/mlx.core.linalg.svd.html | 14 ++--- .../_autosummary/mlx.core.linalg.tri_inv.html | 12 ++-- .../_autosummary/mlx.core.linspace.html | 12 ++-- .../python/_autosummary/mlx.core.load.html | 18 +++--- .../python/_autosummary/mlx.core.log.html | 10 +-- .../python/_autosummary/mlx.core.log10.html | 10 +-- .../python/_autosummary/mlx.core.log1p.html | 10 +-- .../python/_autosummary/mlx.core.log2.html | 10 +-- .../_autosummary/mlx.core.logaddexp.html | 10 +-- .../_autosummary/mlx.core.logcumsumexp.html | 16 ++--- .../_autosummary/mlx.core.logical_and.html | 10 +-- .../_autosummary/mlx.core.logical_not.html | 10 +-- .../_autosummary/mlx.core.logical_or.html | 10 +-- .../_autosummary/mlx.core.logsumexp.html | 14 ++--- .../python/_autosummary/mlx.core.matmul.html | 10 +-- .../python/_autosummary/mlx.core.max.html | 14 ++--- .../python/_autosummary/mlx.core.maximum.html | 10 +-- .../python/_autosummary/mlx.core.mean.html | 14 ++--- .../_autosummary/mlx.core.meshgrid.html | 16 ++--- .../mlx.core.metal.device_info.html | 12 ++-- .../mlx.core.metal.is_available.html | 10 +-- .../mlx.core.metal.start_capture.html | 12 ++-- .../mlx.core.metal.stop_capture.html | 10 +-- .../python/_autosummary/mlx.core.min.html | 14 ++--- .../python/_autosummary/mlx.core.minimum.html | 10 +-- .../_autosummary/mlx.core.moveaxis.html | 14 ++--- .../_autosummary/mlx.core.multiply.html | 10 +-- .../_autosummary/mlx.core.nan_to_num.html | 16 ++--- .../_autosummary/mlx.core.negative.html | 10 +-- .../_autosummary/mlx.core.new_stream.html | 8 +-- .../_autosummary/mlx.core.not_equal.html | 10 +-- .../python/_autosummary/mlx.core.ones.html | 12 ++-- .../_autosummary/mlx.core.ones_like.html | 10 +-- .../python/_autosummary/mlx.core.outer.html | 10 +-- .../python/_autosummary/mlx.core.pad.html | 12 ++-- .../_autosummary/mlx.core.partition.html | 14 ++--- .../python/_autosummary/mlx.core.power.html | 10 +-- .../python/_autosummary/mlx.core.prod.html | 14 ++--- .../_autosummary/mlx.core.put_along_axis.html | 12 ++-- .../_autosummary/mlx.core.quantize.html | 18 +++--- .../mlx.core.quantized_matmul.html | 18 +++--- .../python/_autosummary/mlx.core.radians.html | 10 +-- .../mlx.core.random.bernoulli.html | 16 ++--- .../mlx.core.random.categorical.html | 16 ++--- .../_autosummary/mlx.core.random.gumbel.html | 12 ++-- .../_autosummary/mlx.core.random.key.html | 12 ++-- .../_autosummary/mlx.core.random.laplace.html | 16 ++--- .../mlx.core.random.multivariate_normal.html | 12 ++-- .../_autosummary/mlx.core.random.normal.html | 12 ++-- .../mlx.core.random.permutation.html | 14 ++--- .../_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 | 10 +-- .../_autosummary/mlx.core.reciprocal.html | 10 +-- .../_autosummary/mlx.core.remainder.html | 10 +-- .../python/_autosummary/mlx.core.repeat.html | 14 ++--- .../mlx.core.reset_peak_memory.html | 10 +-- .../python/_autosummary/mlx.core.reshape.html | 12 ++-- .../_autosummary/mlx.core.right_shift.html | 10 +-- .../python/_autosummary/mlx.core.roll.html | 14 ++--- .../python/_autosummary/mlx.core.round.html | 12 ++-- .../python/_autosummary/mlx.core.rsqrt.html | 10 +-- .../python/_autosummary/mlx.core.save.html | 12 ++-- .../_autosummary/mlx.core.save_gguf.html | 18 +++--- .../mlx.core.save_safetensors.html | 16 ++--- .../python/_autosummary/mlx.core.savez.html | 12 ++-- .../mlx.core.savez_compressed.html | 12 ++-- .../mlx.core.set_cache_limit.html | 14 ++--- .../mlx.core.set_default_device.html | 10 +-- .../mlx.core.set_default_stream.html | 10 +-- .../mlx.core.set_memory_limit.html | 14 ++--- .../mlx.core.set_wired_limit.html | 14 ++--- .../python/_autosummary/mlx.core.sigmoid.html | 10 +-- .../python/_autosummary/mlx.core.sign.html | 10 +-- .../python/_autosummary/mlx.core.sin.html | 10 +-- .../python/_autosummary/mlx.core.sinh.html | 10 +-- .../python/_autosummary/mlx.core.slice.html | 14 ++--- .../_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 | 16 ++--- .../python/_autosummary/mlx.core.sqrt.html | 10 +-- .../python/_autosummary/mlx.core.square.html | 10 +-- .../python/_autosummary/mlx.core.squeeze.html | 12 ++-- .../python/_autosummary/mlx.core.stack.html | 14 ++--- .../python/_autosummary/mlx.core.std.html | 16 ++--- .../_autosummary/mlx.core.stop_gradient.html | 10 +-- .../python/_autosummary/mlx.core.stream.html | 10 +-- .../_autosummary/mlx.core.subtract.html | 10 +-- .../python/_autosummary/mlx.core.sum.html | 14 ++--- .../_autosummary/mlx.core.swapaxes.html | 14 ++--- .../_autosummary/mlx.core.synchronize.html | 10 +-- .../python/_autosummary/mlx.core.take.html | 14 ++--- .../mlx.core.take_along_axis.html | 12 ++-- .../python/_autosummary/mlx.core.tan.html | 10 +-- .../python/_autosummary/mlx.core.tanh.html | 10 +-- .../_autosummary/mlx.core.tensordot.html | 12 ++-- .../python/_autosummary/mlx.core.tile.html | 12 ++-- .../python/_autosummary/mlx.core.topk.html | 14 ++--- .../python/_autosummary/mlx.core.trace.html | 16 ++--- .../_autosummary/mlx.core.transpose.html | 12 ++-- .../python/_autosummary/mlx.core.tri.html | 16 ++--- .../python/_autosummary/mlx.core.tril.html | 12 ++-- .../python/_autosummary/mlx.core.triu.html | 12 ++-- .../_autosummary/mlx.core.unflatten.html | 14 ++--- .../_autosummary/mlx.core.value_and_grad.html | 14 ++--- .../python/_autosummary/mlx.core.var.html | 16 ++--- .../python/_autosummary/mlx.core.view.html | 10 +-- .../python/_autosummary/mlx.core.vjp.html | 16 ++--- .../python/_autosummary/mlx.core.vmap.html | 14 ++--- .../python/_autosummary/mlx.core.where.html | 10 +-- .../python/_autosummary/mlx.core.zeros.html | 12 ++-- .../_autosummary/mlx.core.zeros_like.html | 10 +-- .../mlx.nn.average_gradients.html | 12 ++-- .../python/_autosummary/mlx.nn.quantize.html | 16 ++--- .../_autosummary/mlx.nn.value_and_grad.html | 10 +-- .../mlx.optimizers.clip_grad_norm.html | 14 ++--- .../_autosummary/mlx.utils.tree_flatten.html | 16 ++--- .../_autosummary/mlx.utils.tree_map.html | 16 ++--- .../mlx.utils.tree_map_with_path.html | 12 ++-- .../_autosummary/mlx.utils.tree_reduce.html | 8 +-- .../mlx.utils.tree_unflatten.html | 12 ++-- .../python/_autosummary/stream_class.html | 8 +-- docs/build/html/python/array.html | 12 ++-- docs/build/html/python/cuda.html | 8 +-- docs/build/html/python/data_types.html | 8 +-- .../html/python/devices_and_streams.html | 8 +-- docs/build/html/python/distributed.html | 8 +-- docs/build/html/python/export.html | 12 ++-- docs/build/html/python/fast.html | 8 +-- docs/build/html/python/fft.html | 8 +-- docs/build/html/python/linalg.html | 8 +-- docs/build/html/python/memory_management.html | 8 +-- docs/build/html/python/metal.html | 8 +-- docs/build/html/python/nn.html | 8 +-- .../python/nn/_autosummary/mlx.nn.ALiBi.html | 8 +-- .../nn/_autosummary/mlx.nn.AvgPool1d.html | 16 ++--- .../nn/_autosummary/mlx.nn.AvgPool2d.html | 16 ++--- .../nn/_autosummary/mlx.nn.AvgPool3d.html | 16 ++--- .../nn/_autosummary/mlx.nn.BatchNorm.html | 20 +++--- .../python/nn/_autosummary/mlx.nn.CELU.html | 8 +-- .../python/nn/_autosummary/mlx.nn.Conv1d.html | 26 ++++---- .../python/nn/_autosummary/mlx.nn.Conv2d.html | 26 ++++---- .../python/nn/_autosummary/mlx.nn.Conv3d.html | 24 ++++---- .../_autosummary/mlx.nn.ConvTranspose1d.html | 26 ++++---- .../_autosummary/mlx.nn.ConvTranspose2d.html | 26 ++++---- .../_autosummary/mlx.nn.ConvTranspose3d.html | 26 ++++---- .../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 | 8 +-- .../nn/_autosummary/mlx.nn.Embedding.html | 14 ++--- .../python/nn/_autosummary/mlx.nn.GELU.html | 8 +-- .../python/nn/_autosummary/mlx.nn.GLU.html | 12 ++-- .../python/nn/_autosummary/mlx.nn.GRU.html | 16 ++--- .../nn/_autosummary/mlx.nn.GroupNorm.html | 20 +++--- .../nn/_autosummary/mlx.nn.HardShrink.html | 8 +-- .../nn/_autosummary/mlx.nn.HardTanh.html | 8 +-- .../nn/_autosummary/mlx.nn.Hardswish.html | 8 +-- .../nn/_autosummary/mlx.nn.InstanceNorm.html | 16 ++--- .../python/nn/_autosummary/mlx.nn.LSTM.html | 16 ++--- .../nn/_autosummary/mlx.nn.LayerNorm.html | 18 +++--- .../nn/_autosummary/mlx.nn.LeakyReLU.html | 8 +-- .../python/nn/_autosummary/mlx.nn.Linear.html | 16 ++--- .../nn/_autosummary/mlx.nn.LogSigmoid.html | 8 +-- .../nn/_autosummary/mlx.nn.LogSoftmax.html | 8 +-- .../nn/_autosummary/mlx.nn.MaxPool1d.html | 16 ++--- .../nn/_autosummary/mlx.nn.MaxPool2d.html | 16 ++--- .../nn/_autosummary/mlx.nn.MaxPool3d.html | 16 ++--- .../python/nn/_autosummary/mlx.nn.Mish.html | 8 +-- .../nn/_autosummary/mlx.nn.Module.apply.html | 10 +-- .../mlx.nn.Module.apply_to_modules.html | 10 +-- .../_autosummary/mlx.nn.Module.children.html | 8 +-- .../nn/_autosummary/mlx.nn.Module.eval.html | 8 +-- .../mlx.nn.Module.filter_and_map.html | 10 +-- .../nn/_autosummary/mlx.nn.Module.freeze.html | 16 ++--- .../mlx.nn.Module.leaf_modules.html | 8 +-- .../mlx.nn.Module.load_weights.html | 14 ++--- .../_autosummary/mlx.nn.Module.modules.html | 8 +-- .../mlx.nn.Module.named_modules.html | 8 +-- .../mlx.nn.Module.parameters.html | 8 +-- .../mlx.nn.Module.save_weights.html | 10 +-- .../_autosummary/mlx.nn.Module.set_dtype.html | 10 +-- .../nn/_autosummary/mlx.nn.Module.state.html | 8 +-- .../nn/_autosummary/mlx.nn.Module.train.html | 12 ++-- .../mlx.nn.Module.trainable_parameters.html | 8 +-- .../_autosummary/mlx.nn.Module.training.html | 8 +-- .../_autosummary/mlx.nn.Module.unfreeze.html | 16 ++--- .../nn/_autosummary/mlx.nn.Module.update.html | 14 ++--- .../mlx.nn.Module.update_modules.html | 14 ++--- .../mlx.nn.MultiHeadAttention.html | 26 ++++---- .../python/nn/_autosummary/mlx.nn.PReLU.html | 8 +-- .../mlx.nn.QuantizedEmbedding.html | 20 +++--- .../_autosummary/mlx.nn.QuantizedLinear.html | 22 +++---- .../nn/_autosummary/mlx.nn.RMSNorm.html | 14 ++--- .../python/nn/_autosummary/mlx.nn.RNN.html | 16 ++--- .../python/nn/_autosummary/mlx.nn.ReLU.html | 8 +-- .../python/nn/_autosummary/mlx.nn.ReLU2.html | 8 +-- .../python/nn/_autosummary/mlx.nn.ReLU6.html | 8 +-- .../python/nn/_autosummary/mlx.nn.RoPE.html | 18 +++--- .../python/nn/_autosummary/mlx.nn.SELU.html | 8 +-- .../nn/_autosummary/mlx.nn.Sequential.html | 10 +-- .../python/nn/_autosummary/mlx.nn.SiLU.html | 8 +-- .../nn/_autosummary/mlx.nn.Sigmoid.html | 8 +-- .../mlx.nn.SinusoidalPositionalEncoding.html | 22 +++---- .../nn/_autosummary/mlx.nn.Softmax.html | 8 +-- .../nn/_autosummary/mlx.nn.Softmin.html | 8 +-- .../nn/_autosummary/mlx.nn.Softplus.html | 8 +-- .../nn/_autosummary/mlx.nn.Softshrink.html | 8 +-- .../nn/_autosummary/mlx.nn.Softsign.html | 8 +-- .../python/nn/_autosummary/mlx.nn.Step.html | 10 +-- .../python/nn/_autosummary/mlx.nn.Tanh.html | 8 +-- .../nn/_autosummary/mlx.nn.Transformer.html | 24 ++++---- .../nn/_autosummary/mlx.nn.Upsample.html | 16 ++--- .../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 | 10 +-- .../nn/_autosummary/mlx.nn.init.normal.html | 14 ++--- .../nn/_autosummary/mlx.nn.init.uniform.html | 14 ++--- .../_autosummary_functions/mlx.nn.celu.html | 8 +-- .../nn/_autosummary_functions/mlx.nn.elu.html | 8 +-- .../_autosummary_functions/mlx.nn.gelu.html | 8 +-- .../mlx.nn.gelu_approx.html | 8 +-- .../mlx.nn.gelu_fast_approx.html | 8 +-- .../nn/_autosummary_functions/mlx.nn.glu.html | 12 ++-- .../mlx.nn.hard_shrink.html | 8 +-- .../mlx.nn.hard_tanh.html | 8 +-- .../mlx.nn.hardswish.html | 8 +-- .../mlx.nn.leaky_relu.html | 8 +-- .../mlx.nn.log_sigmoid.html | 8 +-- .../mlx.nn.log_softmax.html | 8 +-- .../mlx.nn.losses.binary_cross_entropy.html | 14 ++--- .../mlx.nn.losses.cosine_similarity_loss.html | 16 ++--- .../mlx.nn.losses.cross_entropy.html | 16 ++--- .../mlx.nn.losses.gaussian_nll_loss.html | 16 ++--- .../mlx.nn.losses.hinge_loss.html | 12 ++-- .../mlx.nn.losses.huber_loss.html | 14 ++--- .../mlx.nn.losses.kl_div_loss.html | 14 ++--- .../mlx.nn.losses.l1_loss.html | 12 ++-- .../mlx.nn.losses.log_cosh_loss.html | 12 ++-- .../mlx.nn.losses.margin_ranking_loss.html | 14 ++--- .../mlx.nn.losses.mse_loss.html | 12 ++-- .../mlx.nn.losses.nll_loss.html | 14 ++--- .../mlx.nn.losses.smooth_l1_loss.html | 14 ++--- .../mlx.nn.losses.triplet_loss.html | 20 +++--- .../_autosummary_functions/mlx.nn.mish.html | 8 +-- .../_autosummary_functions/mlx.nn.prelu.html | 8 +-- .../_autosummary_functions/mlx.nn.relu.html | 8 +-- .../_autosummary_functions/mlx.nn.relu2.html | 8 +-- .../_autosummary_functions/mlx.nn.relu6.html | 8 +-- .../_autosummary_functions/mlx.nn.selu.html | 8 +-- .../mlx.nn.sigmoid.html | 8 +-- .../_autosummary_functions/mlx.nn.silu.html | 8 +-- .../mlx.nn.softmax.html | 8 +-- .../mlx.nn.softmin.html | 8 +-- .../mlx.nn.softplus.html | 8 +-- .../mlx.nn.softshrink.html | 10 +-- .../_autosummary_functions/mlx.nn.step.html | 10 +-- .../_autosummary_functions/mlx.nn.tanh.html | 8 +-- docs/build/html/python/nn/functions.html | 8 +-- docs/build/html/python/nn/init.html | 8 +-- docs/build/html/python/nn/layers.html | 8 +-- docs/build/html/python/nn/losses.html | 8 +-- docs/build/html/python/nn/module.html | 8 +-- docs/build/html/python/ops.html | 8 +-- docs/build/html/python/optimizers.html | 8 +-- .../_autosummary/mlx.optimizers.AdaDelta.html | 16 ++--- .../mlx.optimizers.Adafactor.html | 28 ++++----- .../_autosummary/mlx.optimizers.Adagrad.html | 14 ++--- .../_autosummary/mlx.optimizers.Adam.html | 18 +++--- .../_autosummary/mlx.optimizers.AdamW.html | 20 +++--- .../_autosummary/mlx.optimizers.Adamax.html | 16 ++--- .../_autosummary/mlx.optimizers.Lion.html | 16 ++--- .../mlx.optimizers.MultiOptimizer.html | 14 ++--- .../_autosummary/mlx.optimizers.Muon.html | 20 +++--- ....optimizers.Optimizer.apply_gradients.html | 14 ++--- .../mlx.optimizers.Optimizer.init.html | 12 ++-- .../mlx.optimizers.Optimizer.state.html | 8 +-- .../mlx.optimizers.Optimizer.update.html | 12 ++-- .../_autosummary/mlx.optimizers.RMSprop.html | 16 ++--- .../_autosummary/mlx.optimizers.SGD.html | 20 +++--- .../mlx.optimizers.cosine_decay.html | 16 ++--- .../mlx.optimizers.exponential_decay.html | 14 ++--- .../mlx.optimizers.join_schedules.html | 14 ++--- .../mlx.optimizers.linear_schedule.html | 16 ++--- .../mlx.optimizers.step_decay.html | 16 ++--- .../python/optimizers/common_optimizers.html | 8 +-- .../html/python/optimizers/optimizer.html | 8 +-- .../html/python/optimizers/schedulers.html | 8 +-- docs/build/html/python/random.html | 8 +-- docs/build/html/python/transforms.html | 8 +-- docs/build/html/python/tree_utils.html | 8 +-- docs/build/html/search.html | 8 +-- docs/build/html/searchindex.js | 2 +- docs/build/html/usage/compile.html | 8 +-- docs/build/html/usage/distributed.html | 8 +-- docs/build/html/usage/export.html | 8 +-- .../build/html/usage/function_transforms.html | 12 ++-- docs/build/html/usage/indexing.html | 10 +-- .../html/usage/launching_distributed.html | 8 +-- docs/build/html/usage/lazy_evaluation.html | 10 +-- docs/build/html/usage/numpy.html | 14 ++--- docs/build/html/usage/quick_start.html | 8 +-- docs/build/html/usage/saving_and_loading.html | 10 +-- docs/build/html/usage/unified_memory.html | 8 +-- docs/build/html/usage/using_streams.html | 8 +-- 535 files changed, 3196 insertions(+), 3155 deletions(-) diff --git a/docs/build/html/.buildinfo b/docs/build/html/.buildinfo index 5b8f4b011..cf982d179 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: 4c95f050a6b931bc41e8ead0c39126bd +config: a85b8c51cc3365be6cea33d5b2a3c4c7 tags: 645f666f9bcd5a90fca523b33c5a78b7 diff --git a/docs/build/html/_static/documentation_options.js b/docs/build/html/_static/documentation_options.js index 90a3fb8b2..66acb591c 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.29.2', + VERSION: '0.29.3', LANGUAGE: 'en', COLLAPSE_INDEX: false, BUILDER: 'html', diff --git a/docs/build/html/cpp/ops.html b/docs/build/html/cpp/ops.html index 63045ea68..eb0c35077 100644 --- a/docs/build/html/cpp/ops.html +++ b/docs/build/html/cpp/ops.html @@ -8,7 +8,7 @@ - Operations — MLX 0.29.2 documentation + Operations — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -1078,6 +1078,7 @@ document.write(`
  • argpartition()
  • topk()
  • topk()
    • +
  • logcumsumexp()
  • logcumsumexp()
  • logsumexp()
  • logsumexp()
    • @@ -1175,9 +1176,13 @@ document.write(`
  • softmax()
  • softmax()
  • power()
    • +
  • cumsum()
  • cumsum()
    • +
  • cumprod()
  • cumprod()
    • +
  • cummax()
  • cummax()
    • +
  • cummin()
  • cummin()
  • conv_general()
  • conv_general()
    • @@ -2154,10 +2159,16 @@ document.write(`

    Returns topk elements of the array along a given axis.

    +
    +
    +array logcumsumexp(const array &a, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    +

    Cumulative logsumexp of an array.

    +
    +
    array logcumsumexp(const array &a, int axis, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    -

    Cumulative logsumexp of an array.

    +

    Cumulative logsumexp of an array along the given axis.

    @@ -2769,28 +2780,52 @@ document.write(`

    Raise elements of a to the power of b element-wise.

    +
    +
    +array cumsum(const array &a, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    +

    Cumulative sum of an array.

    +
    +
    array cumsum(const array &a, int axis, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    -

    Cumulative sum of an array.

    +

    Cumulative sum of an array along the given axis.

    +
    + +
    +
    +array cumprod(const array &a, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    +

    Cumulative product of an array.

    array cumprod(const array &a, int axis, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    -

    Cumulative product of an array.

    +

    Cumulative product of an array along the given axis.

    +
    + +
    +
    +array cummax(const array &a, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    +

    Cumulative max of an array.

    array cummax(const array &a, int axis, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    -

    Cumulative max of an array.

    +

    Cumulative max of an array along the given axis.

    +
    + +
    +
    +array cummin(const array &a, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    +

    Cumulative min of an array.

    array cummin(const array &a, int axis, bool reverse = false, bool inclusive = true, StreamOrDevice s = {})#
    -

    Cumulative min of an array.

    +

    Cumulative min of an array along the given axis.

    @@ -3316,6 +3351,7 @@ document.write(`
  • argpartition()
  • topk()
  • topk()
    • +
  • logcumsumexp()
  • logcumsumexp()
  • logsumexp()
  • logsumexp()
    • @@ -3413,9 +3449,13 @@ document.write(`
  • softmax()
  • softmax()
  • power()
    • +
  • cumsum()
  • cumsum()
    • +
  • cumprod()
  • cumprod()
    • +
  • cummax()
  • cummax()
    • +
  • cummin()
  • cummin()
  • conv_general()
  • conv_general()
    • diff --git a/docs/build/html/dev/custom_metal_kernels.html b/docs/build/html/dev/custom_metal_kernels.html index 45e410615..6821a01e2 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.29.2 documentation + Custom Metal Kernels — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/dev/extensions.html b/docs/build/html/dev/extensions.html index 544c8729d..1e4132779 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.29.2 documentation + Custom Extensions in MLX — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/dev/metal_debugger.html b/docs/build/html/dev/metal_debugger.html index 12f620a5c..f017c9085 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.29.2 documentation + Metal Debugger — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/dev/mlx_in_cpp.html b/docs/build/html/dev/mlx_in_cpp.html index d1efc9087..a6f412b11 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.29.2 documentation + Using MLX in C++ — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -139,8 +139,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/examples/linear_regression.html b/docs/build/html/examples/linear_regression.html index a375443da..ab3176005 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.29.2 documentation + Linear Regression — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/examples/llama-inference.html b/docs/build/html/examples/llama-inference.html index ad3b3af69..766f5bd40 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.29.2 documentation + LLM inference — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/examples/mlp.html b/docs/build/html/examples/mlp.html index 0c1160678..37aea8d6f 100644 --- a/docs/build/html/examples/mlp.html +++ b/docs/build/html/examples/mlp.html @@ -8,7 +8,7 @@ - Multi-Layer Perceptron — MLX 0.29.2 documentation + Multi-Layer Perceptron — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/genindex.html b/docs/build/html/genindex.html index a9538efa8..ed4cfbdc1 100644 --- a/docs/build/html/genindex.html +++ b/docs/build/html/genindex.html @@ -7,7 +7,7 @@ - Index — MLX 0.29.2 documentation + Index — MLX 0.29.3 documentation @@ -36,7 +36,7 @@ - + @@ -139,8 +139,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -1245,7 +1245,7 @@ document.write(`
  • cuda_kernel() (in module mlx.core.fast)
    • -
  • cummax (C++ function) +
  • cummax (C++ function), [1]
  • cummax() (array method) @@ -1253,7 +1253,7 @@ document.write(`
  • (in module mlx.core)
    • -
  • cummin (C++ function) +
  • cummin (C++ function), [1]
  • cummin() (array method) @@ -1261,7 +1261,7 @@ document.write(`
  • (in module mlx.core)
    • -
  • cumprod (C++ function) +
  • cumprod (C++ function), [1]
  • cumprod() (array method) @@ -1269,7 +1269,7 @@ document.write(`
  • (in module mlx.core)
    • -
  • cumsum (C++ function) +
  • cumsum (C++ function), [1]
  • cumsum() (array method) @@ -1761,7 +1761,7 @@ document.write(`
  • logaddexp() (in module mlx.core)
    • -
  • logcumsumexp (C++ function) +
  • logcumsumexp (C++ function), [1]
  • logcumsumexp() (array method) diff --git a/docs/build/html/index.html b/docs/build/html/index.html index 9074ae226..a61d11b69 100644 --- a/docs/build/html/index.html +++ b/docs/build/html/index.html @@ -8,7 +8,7 @@ - MLX — MLX 0.29.2 documentation + MLX — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -142,8 +142,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/install.html b/docs/build/html/install.html index 0c6c4ee16..fd15a81c4 100644 --- a/docs/build/html/install.html +++ b/docs/build/html/install.html @@ -8,7 +8,7 @@ - Build and Install — MLX 0.29.2 documentation + Build and Install — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/objects.inv b/docs/build/html/objects.inv index 9d125e2fa523324bb750895d35f02e371820c361..7c7cf5103cfc2208bb9b16b9a1cfa28f49564784 100644 GIT binary patch literal 28708 zcmZU(b987i^DbQ5ZclAnr?zd|wr$(CZQHhO_tb80-}iU#THpQdKiNCUBzyL(b4^UbPwY@&yS6}pF}H{6wcB30m}Ilbj`x*S z-|ze#w%zWg&THR|&%4rGox-@g6FlGO_5D7ckNMea{`&17_D}Kd6anv#N7x*nAMVcn zw{V=b8JsG0C@G1`3vqX_bTA_2MckxXzzu5nj!ktnrW>ed`qiHoJYSlBKcjw-DbA#8 zH@Yfh`<2LLE>99+&5#*S5G|UOh`rk~&d!Wuyx;FHxtmAI;VnF$#h?4&JrEa+U!@J|D#!2=*T(Ids@I}oDu6hxqK!v#YO58K8IJOt z5!Tz}&@V~ZlRKk03Ni%9U^SeAnzpLH-)1V)jkW45S8L{2+~o0!(Ekumpi!Sq?iHIxxVm1ra9muOJ zmyM^}^Kjs+JX#>p5_*cXfQY^oKiq3m_%wvFNuJb2@{gG*Q~T7wjIfVCZi{PbtV>(2 zL}JP;`)3YWm8MAQQ&w%{pH}8<)j?BN>$H$8)$7EYQgAkeIU|wxyd2s4X=~3oGH7>% z9HVQVp*I&m=LV9D*4iUhKS;h>fU|^)&+{ay+7nVDE88P-pk4DE?75io5S$Is_sUuIwazjenlwTyM zhZ!|7ziE^AZxu1`WoAatGFrb=P!?d0l&1#1+3w zSg(m;i&g0aRMCYCH(rjdcG6`cyO@l1U%AD=Gu=k+gKLf;y&t#rRKt?u74*V6W60;G z2kNd>^7lqsv0qaU6nQ3b9xPl<;C}|;6G52S1Qd5LGj+mEh9#J9<-`3}mi(jd;nI|f zvoqEd+)b@KNHS2V=DS2m1T+*wStmI46p~#m3?z@U>(7(TRH}p;O-S6nAH~ zeyASP2;5dK6k%ThqJ7g+Pj65gKUGwrxqhCh3}Iz1;NM_RM_@noPjGjmlCAaf#C3%- zf%}5Tm>9>%y1?%1CYR=OzQKY2cc=yRriRzrmD&UUc=U2Ig^w4RqxZ6KI3{uBi$;bu zonzQn>;sm<#*?JwgfDM~3-6h+-){88dv4~nETp!<1Nz189 zd2a$o&LnD;l%KPdp&gXArwXYg zYaWhM2NZGNwLfNjRUQ#6>Af-)pFpU?02gs=Sfg66|T~gjGww_ z$0tlKTD7Yt@8pwb>8M(S^|)~Z0IkqhggjMnzZFtS$xR<;?Lo0cXCmI&qv}zc^J7O) zlkG8EsVW|pnG*u()a3tw*VZSn<~RF7PD2e{+xIH(u{+9q)7OL)rmAaGf$KJQ$T-C| zc1Y0rm^v5tG?+O4J@40zFT;zO`xkj?A}&cy!yC} z{79Z3Z$7588WUae7R_8J38?L>h$a7waUqd5)%91$n|tZ5?#KsI){S(>Q~JcY{^ZA@ zZ2scL@<5#4onIKjc#ynocc<_b&EBtiZ288a*#4}oz2tn>)bP1yNqYng-kq;$!oH-c zyCj0DwOxJgt1!`Jo49GxcAzuT@TQ{~JEKyPbU-$I15G#IzX=$6s$Y)BcizyI3#1eK z63)mtlS4Nm?Z^sF14f6x-;?M%9{6P0JNnz|>k{Ku*C z<+{F%mucTqbP_wwbzA=S-MDOyo|L(^)Ab|8(ai|ILJextc*v3UEdX^*M9Zo-?Vv~J zQqA^8gK|Kod9{}@ycfecY5Db$kb1r6#b4(D6zu?*(HwD;$5f#}`(6T&n!JXad9Pfp zA1T^js?1xa>TISW|H*J>-kiyTqSNEJw!OHUnx#AsY9+yPLEoq_kK-hwe6iLH>heb? zrNs@!OT#%8(4_p;aBk6p&62#!?Q{*#B9kTjQOQY0T(nL6iJUDGdC3dFMMirf`Iy=4 zUaV04NdFb{Df}?HG-r2jRil{wJx}`L^ zfAp++n2ug{9}-xSByUtzOTRpl&UZxSX|FUji?wNbagBR&QY=6=eT2JT4^x+mjAmIc zXFIl*TB9BvCxDu8c0z;iGKBA%%4w`pvnO?~1v2@F)#78K{8n{Qxz%8^pZBpCy&Bo? zFTWyZ%vXWO!vUj!@)+MFhmN^0CyjM)d_o204A*Xe7 z#PY!}pb2KGUXBTAL6|aNUreqPGn>q#E}s{mqzPrxZZ|=6WP{G=R3F)z7C5mAmt>*l9{d8a64*Y`&fe5+y`_fSW=2|;`_)ZVm-6!ftvgeZWse3!kl2k1zl z1Bp|tu_l;8H>;C;9v@EtXZba~nQnsgQQ|!z zFjAMaC;TLayEVR&bKviwqlaW8|uR zWN&A;_pQPv=vqoy>-K8tgXDgvBqR;*%tYxcMJ~9eD*SiyS-rP8>yUqMgGgwtOku4^ z?E4($8!G0hYUk~^r|od)hZNaMCQ?(^W(G$!i}&T5s(SGAi#A{?&5f=;3%1sosyh6} z^LK3>Q0R{^)`4pc_Z|Dtsv2l%_;Y2|#4%9$t}_kKcTMxpx%>+Ie}|aAk?2 z_{60JxLpsaTO;o4h>Kwt-j0|YCOMrbp7Fu$$Wqnl;{fqyec;vY0o6sI@=PM^a7~Z$ zKUPZULzjho`r`GJQ=1;*))=y#@ZO{fi^-Lf@4+jr7A+{#N8{mkkh^3`$o7Il+Gx{! z(0NE18}P2jm+O-V$uD<^W5yz)H%>I{X_b4Y$=AY~xcT#I#z_Eo%6!uxsx; z9whCoh@C)*@lksU`&1&IUk|%vkLu8szrVy_d4enSNqV^|ST{*|owBn0TFIqh@kN-t zbV?sd<$0To)CRNa9oQaJ zGJVB+eOI_|IP7T)?C5F#(_itR!bfW)YF+Rab>J1B{25OwnE}#gYtjD-XKU!~VU z3CtQ7W6y5S8kbQ~Ghbb_X+pNEL#yU`YQT2x&_Qc&R`mvvvo1GYnC>=0dUggl@+H-+ zBGsv**x70>e_=NQ%ym0r9d9Frf+fX4L0=_~>u$JE|Uu#7}LUD_YEwnc$t$Q-+)%VVj{o>g>vv zZb3+tE)IL~;#4(9#La|asZ=$tu2F$%*{++oXe|0xcIOmzrj_nExcEO_p7M6W2JrE< zT=s^)-%eGqjdLZC!;GYTK(jbH4Od_Pu79C=>>vm27+tU`a2IdI5W1-vZhck|@krz1~M+7Q{e4ox26)M6>nqD zrwV7*5$Pj2S90qgG%2QYG_#j~L5x%89X^|-U_)JMIIBC-T8uiFKJ^bhDlN_AUM3`M z=fLfGN&ho}wKHzcq**vd3#q>-g=>@CPcr07Qq@LtaSU#3&ZYWZa8rCHp7N|=#h>zO zu~0Nc5M9PAP=|CGpqy?9G_<~?wO$dvUhm?Pa zxDm_Fn+u|u{@s2v<)eD}7~r&lRr^pWX(@sBpO^Nl3G8cxa%Rhpl4vJ1^ln5hLuBuc zF5aDLG%s3xmQIi_(gf3*@SGxvLAf+)Fw4 zN9s8Qe<%M12`D+9s*VJ>q6ceB3VD@UiQT%_;wQJ&l`<-0XI$0P%r?nwP!k2R3spae zzG&pL0Xs*I1Ri0-(o&|QqMA(Ww|*0-S-M~{u7_e6<9k9Nwn?DdWq z=vARl8BARNk?n`N7p8nQ`U{+djQXo9l~G_l70&oBh%~?uh2a8{jhaa$LTMCcw^kkQ z%H)=T41fuKPhRkA0NO#M(m!}KZ_OTBF#Pd^g&}dX9rNRCzapjqG2>T-*%8oFt5N;(f#h;5 zwTei$DCTFmp|Dby>_P;TVPzaj=!-)gT?0I~Jn_$(*tQvtmkJYl-6%_DGmmaVjMq5F zz3|2B?$!I-gAzI45dF#+*rB_{#gA>%%>S{Kw&(xfR>qWOf6=*QrBjp;9Fwj1a+Zy> z!?Zm+WNh?^oQYg+{!VN4;jUV(y#HtR|7cZS7){2@oCuNB`sFC*g;?}6N}5nv3P8?3 z4uhZza`w|)w(!qq$`U{z`NZ#)kx#`rgBhw^;eSMD3ls`iA(1W|pzWGIbX>Mjo7OmX zlMB6+{2``Rfo#ygcbGCW2?S6&|0@2uR(F4uRR8`rcrCwqCkcWbKC{_t%f1`kT&vsa z{d&<$#gGbOHevrEym5m#FuG|6`8>L@AP;Pisc^%kR+&S0{4fdjv%!39dl6~bSk3Fh zJ#l4GzT>&ZJ=8ulrYYo4XY9W5x`=a~Y#n>MGZu?XH@`4xerM*$TLjrX&70ET{f1=y zjd7=7eNzQGy)f;m+4feD@^#PT-sq5tyWwDWH#SDlB)JbgeIJ$>msIG+Y>X$E6IkL^ zcE>*P^Z-Y0srjZ&%H9NfG+h38 zI{I%P57>5abi_7<&|mDRw?0!XNAc|KsPaxgcUlTT7q8q1nIqg%q@<5-$h_%^ zUq7&#UOW{qu|9LS-c`!CWY|QUxfi8H4Z16-n0k~G>ECb6Ga2GJV%=++=Uh_8SY$@+ zP;c8bu&TG;qVhnnEU_-gDmP&Br&YZl0iIBnzc&A7AOemK4kxquy}-0>{zFXmSges6 zwL={-!y&$c+w1Lf;Xe*fojqKSuky}FV0UtflxJ++KUaNo6~ppY7eKTL?fz)iTVT7F z$_?njROo%0kVXaXNd{1Hz&dY7UMNfR5K;4#mWcy5s!jb0O*tJoo|T{)DUeG!onR!+ z2&%`R>rCg6BhKbijT9R++yqP4j^oZ@d_IjDOD-5|+mz7qmAW~~v|GKz_#1~RE7qSO z?f!ATIZ!htk_j+os^u&|GNmMmB+eFVG;o`gnvVoZ)?QqV%8(V%Ql-UN@o59_HEOUm zh?R|O7-?By3c4RWel{;-cPqzPb|z!v{%#-mdTM1^fvvUzB;yC#XVRP1w+1a}%olHq zkK;@CEys*P|PAf&( z2X!yu0_gSk%+pVVh0J5?h5DHr(J)83toz%~8SoohV zbp73W*`j|y$Gh1k>GCjQrF8?{Be9D#>W-!4CfwE&C*GL@fAe1Ni=a2YvM+y7m$&}r zxfOj+#?cxDHx#t*H8@nf5HUwCFHJvjtiIZJkXYWFoU1PWSD~%!8+0=5xtm+9pW3pM z8<`jXcRIn2C{$`XnSX(nfT3SD@LKdTi%#}75M z#mgo!J!eZji9I0Zr$OAmA>T4-G_3PqcQq+`{J#sD)fpz!0W8R2xDiyX5qm9C`;Q78 z6MYkFlix2R*u41BxYD?U+QAjX+D75~0;8tdyrH+#s7I#eDzgo3UlJV3j^dQ_Z^uZec4S$}KF07{MVy1eH zJ&8~y_FzV4Z|Y**;L&RnN#oy(J2`mmOkV5(pZf$cFPOabh}osR;YDJ%OV^9C`az!| zx(h?)?(g6Zie^BT=7Bo%V(ilV#P0=ZS=;}OhoRoJrCm5Cd*SnK3EriWVNFQBN#wF_%MF!g2LOj?kW z-5p|O92}WKMcwMl^SUnPyBrPfkp{BohS=@m*feE`ddIF>Hf^j^d-;v$E%$_18GMkb-=MdpP=f80X#0m@eb!w>cq~%ci1;u&gKC2cWj=?) z1ar?g#(nT6GLnHHC}vCMv(7Xlbry$Z5#5f0-;Jvj|sbdghOGdJP!u=&uVrj!!6*3qH zDo~Q@=$4dMA-&8yL4aj=0nQ8ooTFH2$RlPc4*-=|PV$npW&<9*?kjlJ3dzf>R0Ds) z#Upi+3$xH1jLlT^*zceFf}-^s9MvP?m9ip{st&+rSA%U={CCx}t^VGif!2GN3yWxj z6=VAo;OwL~{(pG|0hmK2HvvS$FRw`F_R6Xxt1oFk{| zEAqHF!0HX)h?I1HlCh$h3V^5TO;X%T>x@472Y9Q_TAN!_R~&nHd~IW2A}FDgPEfR+ z&kRxIf}1BYu14C_nI9xVI)-@x6cO!zB8d8ZN2oCG4AyA>-FS-1ig-rWjo(SSP`fWN zy(yLayQW-dwsv%CNwz7RE7ph+;Fn=_-Ve@Z{k>gi&pzAT<8QYC2Mm%uftZKBAuFF_a;MIZcri~D3rouSp-}~F0Sq!e%7yz50T%pVI(IKy5e%Suc z=?M=2yZgZuwj`bJf|_4~g96P1Vr^U@mT^Nf>fgI}dk43Kg?6_|HQe2|h~}z6@p8hk z{sOMUX&h?pu$lvI$?_V03sgayR~Qo zXD@9d*cerpy7WgmE5fS0K(%Y{(m`4f#h87$9kZE?Da5H2r6Lnk%L6~D7cG*&Q5R)8b$fV{ek1$k!T=XMp%_kJRn;>VM+AC8-4Cm`< zMcji014ycE7xFTT^?7^9E1}sAYWCK;V-5Rued#VFQgEuS2l9#J8(I5Hn<05>MgC=S zEU{{S{0a@>aM|z`X4g^)d*ks1*lg*cgJ&%#B8_TZUchlw-<$cUvOKN}V55`z;W4m&?yncpu9&qpb;vRUl zuAd1nO^X2Ep!gMW8n880p)MAW%4scS415q-E<1X^N-f?{vxHt$T!Z-Remh>UYL`o^ z8lep7bYilP{DHM+fg!65J0L}%5XV=@;CJC9oU$k+IvU-aAi={wj7*NTR z7)Po2gk$fNy6?&*= z0dOYOdv#q^nSli)BDp1(B_xwFZq6oeI6mO6v2^{;&RFv>jUk8-$Q=*d3yIi#ub0S? zb-asij;7q}JReNZLRo{wiR=<+sHL%_dEQHH7RZ^anh(?NRFZyW#WMr1!&9NR8BC>jBqYaI)uq@9iK zC6uynkyhXew9J6zRlhm;J0ldOl&f^8a${1!%Lr?2zUD3Vf9??T9$i`3f)|T9734x9 z`!lvu;c;a3@@o#OY)I878Wz|M%11Rtze74_Ef$`*#A5_(EgjH<$566v{{p=45FhP; zI`D#Gvg3m(`gPOUojt5JD1ywKG0nln_)c8jq7cQ((=fq_)2qcq;P`EiA4)^NaZCqp7K9AB1wzVumDm>^g+HGy^#I|+^M}ui07=g zUwc=JhuwsX7Ia}-RU~MhBR?v>Xj;jr@esr-njXS;&b%Te>-c7(P{L`RV=ofilwy8&9o7Zx>mo(-bJH5LVEgTg^)J^n5|1+roFE^DxWc1))rGB%sMR;F5zflh&J z)Tvx$&c-7>8My>vvUSBHl&xLV`W zzeSX5_M0|G#&qe`{5HR)$(#furF5b*5Axw5U!J4qL?pscAe@rzMXdBqVdT%*(Nzyj z-__RcZ7$%8?y65PE)@M7A$V^`W(g!hJQ_tS-^=|?bJ}tvfF-b=!3FwH47O~-7UZEj z<7-U#>G2Ccih|u?B3aX^g3)k--?9@$>rc($B)K^k&o`j``|>#iXXo(4ED)JzeD_o* z`n047kFbPfTB;$vC?$cBtEdXo{wM7^6sgpnsrHG}U8{^Tv)~iMitBUtJ zqQc(r7=I!SeNENl+Tc+LxXuyjm&QGEyKYsdIS<~AsAxS+66e#+Gofqp4e8d?Fb*M; zekmxF>TkU85_C*(XzaNF`?d){n}UjI$#bP$85@umCwhH#sR2-0w|)@4V<0tY01nKC ztj44M;iOOlEP6Wum=b&{vM~nFjwO$R1sUg@iU!4KQ7SDJVe4!R*I!)5M%t6_YW>Pz zNygs(qhV#a@zE}$2V*8UYg}_IRVTW^Pp7>Duvq#u#1?{{Z{Mezjo`~7NIB)_=dPTY zy^%e`r9rd{9gE;(dx3pdYb?6=65(wp)QiO8kB31~yL=o?AAtyblsLwY;cYn@od@_D zEDi>_-pfo8R=+Q|!RJ38B34P#yr4b;(pp%Z7h^?5=Q?A)nq3CjRbfW63z_DJbzeT8 z7=~CihYXBZoystvzmliV-v^;IKO^7*WP<-BrBd_Uc@40qguT(YysiEzq8z zJUdFm*~X~FFBR7v~1;~^DZ*=GXa!9AG7G~%rpgV&+_a!tif^`hkTLbhfY7}U1;#l2#mtDX?24P5jm~ESd6HY*R{gkohK;)Y3wGcv8OMg~-~Fgh1vM9zBhD<#P?+Ze_UweoTEIznZWjRLvI!^A^93B)adVZhsI zjo^*cIG{??gU}0+5nu(?y0(~>7f5hN(HW5u5I0=1Hro%-Og3`$dBkq|yr;Z;vgG+m zRO5*Q5E;RK_i?P`{HwnE+k_MJpU48hbpx=g!O!(u{MOuIu%|lkr@wd+4>5F>P?TPB zmFup(bJ@kAi{^ryt?}r-uy4N+g_({%Ik(K6*OvWpzz%c_u&-P$jvN6CT)XiH4a06{ zGIr>VNug@UQJeC%4HjACAGr~VZ1oermy0^imZ*!RkcC5(cV6RVj=o&t(fmKcoKs#T ztbiUpu>KJo4g>?|_MYr6qkzSpsqLW{WSlA_PdIzkMUWbZ$*jBj4xXOj+-pPAB;d`4 z#w6y7=fYzUZK_c#;2e-dN{8PGp7>Uz?7JEQemL#*M{+#9>|I~&NKF1KO|Y09U5F=Q zntGU>JSPtJQ7m&~)5NfaB#<4S5K|{_Z)$S&9Qh1=A*dBUTi${r-3d7UKhY6yn;3Gf zmAKpu`R(_f>d2qQz!NQ7^IGrOXAo^4^Bwt*FreK2*b=dFk2%BBBC8EV|L`V(NvX*m zspOH}-nuHyE%nF{nbd@ZDx&Cxk`Uw22_w!?4I7JHCa%SJmwIhuJjJ_=_cUfeG|_47 zgP(~FKnmL*vz@WM6(S>OwuN+RiF8Y*mShForU|b+D|LY(_i@l7Jx+XHvm|5 z5|?SpFRnd{3yv|_0z~v0R&k!1L~DArWE5|S4TB2HjkV6YdLXB;%s9T0=$8$7*l#j|$ezmJ(Q%;~l}+<%6tq_cfh*0|-ZCls~9bogNBxQDCrO zov&fF+uMh^`5fF8AtS)YA-wNL?lXJGBf^0LOBzYozCqpMEFdB&{a}z#+8oTo*H$UKOC?KEij^Kj=dOYBouPON3!(@IH&st&kIL5xu z-{FgAyVpIp#)fNm%@!=Lb684Z-sTK?rYkAnD@BF$Af4 z_}rg-^6dvgMi45h<mlj$RcvB6G!9u)sG7BZg|EPMV z*H5k{2F}#BhbL}$Cc60Vtv+S6nlbRI!<2|{xBknVZSxEBn&F+QURY z5BD~uN?Yqm^CP+7^wCKn57}}Dy_R?^1A8TJq5H6l2U7~wd-N;DsE%*4dj1;@#E#ME zTxV;^OiOPHuJK&$!${8w8?Y}L&ajQFtxL!*N;1e^ZAR&CdKb(L^x4w8laT@(= z5bvXYA&40KNLXTwAmzUA{(k-oz+t)KkRP)pL0_g%m-4RfJUa~_9b+*25(y&3;zSk_ z%qj3+S}LPp@N3Vd9p^3cjo0wWW#&Y-P{v~7)4d}JwKp#fgg}_2yk8$ao4#@VxZ;u+ z`B(=AW0_HyH%-nV0IX|Vg?LRCvVQ#MfFRcDBW=VFlW#!S!gf8X0>FJg?Y3(+yeaA# zb2CW5x=qJ#yPuLBOjh-4)>H5d zcIN%Q=k;qAi}MV6=8#=P=InD8{+;;HItqVhG<(%CkGL1r0RDE{b>W?1*_0h3{V<=? z;+HBjFvTS}&nwn}UNsp2Gh&gm3`v&~d}4GMG@J9vn`APGIdPCjoOiw>x z3jteWU;64=L(|h~d$p~_d7GW-#%s3yWi`l=2-{Xw24q^J$A>o6Kr&-B?q|(WW~(*5 zwv}0*&U8pLIL8cP9ipq)p;FsbtMga28(gH_JZLlOW`yGvZ@)_t7%P%FuU-~-swglh zuU;GiOF}@8lyzz|TH9i7P52UQ^AbT*w}S>f@jG0w(gsz zuX4^l!U^&n%E=!4DRbdI-!k%@(DI(y3UlEb`vUS8SHij5B6IWp)I9Rt-2C3-0`sAs z>oKyq=kYe~v}cXCwym=%E4I83VU9d5woA@F6m}aBkt+QpDu-RLNR3_Aq#gc6xErB` zUDK~(c6(4!TK&|jXMLzx?NO{;x7k}|4=3ZDjW4o6gD-5QFV<}~Y@0mmw=LZ}RhEKA zt>=&DciN?_nOe7~JzF|l?dV_Q&P?yGJg(5fV@zfCo9|;fub$t5NTY-Nx?4_;JFnyK zeRs>|$M#p7Om(u)p?bBmzc+DrtxA0xc4U2&xM0+=khH~Cb)^s7?oXxyG_jHRnGU=0l53aCMUNv>G z%1BVowCMD1$^fonSU{fwxR7sQi<8eFQVk6?r^Pp9#%p{{t^|JfhYvQ zNI9#dgJ0A-jEV=`Z?9tg*Nr!q5UYSSCla0+`muv)5M|)s&&(E*p{?Co_ixbq0L?6_ zhC3;(X+w0pMjuRxbM*R^b5Z8A`!cA5ZKimLE=EfnCQl2M>B9Nte*S#>5J&3(ij8Z2 zja{XgLHWsF&V)Lu-TUYC{{0qfiW?UzG8k)9(3k1<_6Gkt=4ZM)rxVzj_j@r457Me? z=1#g>trN(RTiA$;@5t%ni0=liWv3Ma(Ra~(v{&Mptl2u9f45aZ5D0hxWq7mHE&H#O z?ol_5eM>CUa8|wxeEuzI{?8%9)lV7-8wWJdd-rIY$0yrEFnG$I?;kC9yQjLz<)6_v zTh|25EXWQ~i-Ka&d>ShG*^5_|$nTP(Ou^=tRSh)@o(S{&V4mq_2xnhbn-trF8Fa?t zeJl*a_B&1Sv80kni5jTy!_i#l?k&!JNWG-JQ{JC(9PEL@J_F5X)dgD!JQ2KPywdYk zwjv~}BflbJ?HYfP^`%m8?I%t?v^?y*==<+Rk%09ubm#WEk@QD2i==wrWG{+Wgkzi) z@>3(2`$shdMzq5za(ldDE^dsHs1bf9k1a4pS$JVaBK?IZu+GslSHrDN^@AU`xHn#r z@*l&L%`VFY0vX0oJ;g`gKkk$KF@;k5>nL|>aZgP`Nh~;X(v?qf+&cvfig9ehR+bSS z4J9tg*i$dE@Y1hD^Fv8sY=HfVjzW46OkhM_>s%acA5)@HfB7*H`l zo4%WSDrQi25A{~Mtez`9?EJmqu?g7p@S48fM)+({l3!CH;*YRdIB;he(p0^>j9<+= ztvzepe}8G_*C-p-yQVtYaCKk3{_F+3sWNNoOIvB1s3f`nz)@A5>c);tGFW~%T8+0%xf6^4UU+ng~M|is1C?pRL-yQygx$aIzM?l z=e1jpWTmuy2p$UpvfXpN+W=p8qMy&vWv-lSd)9jx)}AaUU_jiuKM#wjGkBf4J>70? zB97Njr!Fwj_8v|)J-jZyE}gSIh#z0A4X{xY~xNY;k&Ap}_%O&{vk5o=& z_xrW=r!(-JqWEC+)Kn_ORjyUmvrQ+NH{0iuza^Lch_z#I3s@`Xap0$Meh}bDBb4`h zve#?Tx47Y4#QG<`5z!NVG8<->_blWdb5Himc3bbe{pb3R`hod)Y;}T3K&^`whZ1Zhw?y=m0alM@23fX!(tD^ELQxl!}|x+8#F+Q5#@Zi3Z#iB z56zRHw^qE~?$5`N*3Fw}jMuNF3XXfv3)T{!k3iTNp$I zl<|~CQH|sgiZZ~Gn3G-XrU{K{r!Z(!UHHeChQTo2!oZUS1JJ=xZU`XuBTzXJQKI$t ziYi8c^(;&^LX>T{;nWcc#1jg*p5_DwI!ERi2>l=ri+p8Mi`zsm$nnPt9&$0K^S-R- zkYaj0=y{6tO%-JH8-ZN<`g@VYu}IrfO<@1LRr*fzTuI_fKY;>~Nrh|`eR4vDYWKnQ ziwF@<3jBDfKea@(?$b;ogprzy{qrj+36$iG^a&gvtLa5$A-f1;l2gfq>c}A&(+5n& zcuj=IJ`p_Z;c;R>d8TvU>PH=F3P1w8#=gj*)d&-Tv}s1pkc2YN(}_{})sDD=Hih98 zdkz3+-S;KMA549)M$@?O@g=;Ms`=0#9)zGmdc9}DSSPuqq|biJgF2ssq?B+e;Ae=Y z9A&U&d~n_E5hTz|dJ-s9>r;o$+UQl0(1pc7BeUh-AXLU<$)2Qs@nU=}}2}P8`>$VH>&h?9asU3<)C@r9heKqlyH=g!JX0 zc_kN^qxQWT$y3Y0mBaO@kIS-j$=O%4`1#LL?*{_!0-Epx&dn(v8QU5Kf`w7?A~0v~ z)yBgp4)|0-v4h~Hc+n^!84*SO!BtT}&66k!OoCDB;`yTDp;bUADfCA0gLR;If=bl= z&~V+h@J=t63TvCo8s>IOtjCxdWP z{6qUlXz%GE8JNG?^rFB*)H6%N*!DlUK`irA!R>6#WBkvHd90D(0_f!#Odgc_t@uai zTZHz=%&ED9pohR97KtSTrG&!@_1d>&p=Y>BxX&S(p3I%AqPgh^tI8WJRBwdI_tB6vYW)@DL*~3?JtT zi&w2f@I|jb5Xw9*&xLfNrAyHgI}UaaXo(Z24hoGC0HtDFqDg8m6&g`G0T&P33Tot& zkb!;%3c&`icaG1mMT>wQ%0VnSV3#fsldBvb#GaX-j3BKBSB z<9=+875eqsg=3_=r>oD3u!9w&!MDk%)9CQ&}a17=JGzVOVeKZAGXdxjO$5F>= z%k6kVX-V_~h{s?)VQPV%fQZy$ix}S=l8oJagqSaX3r2yyY0!s)@Js~O0wLkswMO#Q z0nT7EMS$>2tvp^Nr!h53N%OHN{eE`~&e(Ar5bu!y_Pwd*t?`4S2jJaeh@y~))K3S*r*;Nu z$^ui@67PQlZ1^O7jRrza>oTkZgcb76%a$P9ZE7DMWY#i7i%Aq$5u|CvG4PEm$bz)?+j!Ndyvy zNLa-@{0d?b#zjQ50n-oa;4M1>brP%hDP&NA#7KuB%SD)$!}&=V>^f9uF!A6-JNzmkP51yE#4R?=eqX;7$t81xCFS`8kZ+gOr`acOWYR^6)SX8?iRE0F~%3J4eZVHvTx-D+8a%dlCOk9!r-&m4_%)Tro~YDA+|?f z6_;ihA@Ij8Tb@BYYrgXs^@L0esW1!@>~J(4q*kke-4KpmdLB$EQm=02*eCf6vOD7PO-pDjxG>K~_$k7alklN+pPr+Tkdhav`0AiV^37;6cB=$T?cAx8R?8 z75ZZWM9ZYNN*xB+AP8O99cBs3>pL+8R}N?uLRVNv@+-BzaE>P?B24o*LLz|?4127i ze|ejjQrW#4Su``fE<^mNK&LMqyceQqd9K*8D0aKc4RU`RCaHhSmM}ta06~)QBlv1{ z1Hxe<7+R3n8z9(eTBB4iV#?VFVL*3V5i4I@aJtThn1dT*5L8$pJyD;Xf9g9Ialjor zU6BwRVz3Uj36xAj*k5#x0%ABPw-ppQ^#~BPVSSL*X!9m=r68yJFib?LJ>Os-z&__4 z6vsBAv&&;jJv$L*dJJjfIer4@xbj*@u9I*=)j2l!Fg@{*G1@*p%AyEI;+20p@<@H= zC;m}kUs6;kghGO^g}PfO9PA<+h3MUgICCk42G=+7KIL3|CImpSg{TOV6#}vlxTsR5 z!5+mShES{g4ScWT-nWy{_B&C@aS5ej5AnqB zVV z126p06;dE%8Uht2fh8V#gcla*Ux6wDmLOS#CxLDrqGnEWj)|YdDU6n>B8<^VeLs*$ zwM7QSkXQsnFgP1O$M_aepfF5x1j(4CkOK+57(fHVNb26adO}3p50Q_BL5!eXF%J_0 zCxWGc&jK%zNQHe>#A%*JQLYrrF?;4jo5BAm;U|zqDiG!#B9{a>0pmK|q^-Cl2|^7I zJyqlhLLbC=sSb+LEJ9T=42WO}6Hq11@+?AVfU#c0KH@CY-;0PTBEYy1fv?{TMNtqT zh8La)f{1&lZ$QiHl1y(VN|OR3K_P_q@+3ueogmT;5|Pm60GJ0ZNFg!7sKQi&S{Htd zd>&8=xsuzk|E8bfk{-dNCOP+NbiwBm?uS)_;5{nhZfOhvhX#I z-qyuMs+ZG*#6%1=lT`S$C@4{0AX1@U%wby5uc0_P2Eupb+XyOsA{c^AME->zp?aAY zD#R}sq>{slAY3DY6&#&ZMoAQ6GS5VYX)4lv4dF4O1{zoa+as_*H$^gBOD#h6|MqFQ|t&!hRX6u}8&sS@zsTt+fPqJ!x#5E%FReGw5X4s|3f zilRW?#Pk9t241b4kA;U>fx7u44Z3&SSzhEJraHvk zF`6;R5g>a(P|#+lG(;AcraVPd5T*Dt43i@FMHb{DLJ&v6fL9 z`Mii@Dj4^Wq=Z4Boj+nUl@_6gZX9PaW*O0D5i%W2Oe)Edkuz8@hEGJwgKCQ)3y5Ha z4AN8!>!nQ0Qb~~MB`+p1*%4!nG!8Txyb$Bui=`JLo6`}W4!Rj}WiN$Lkn7bg#^ZvA z7)}U(WRh})-%?2hlL+5qhW}}A1pgB&dM<_gXE{8vt48A>>w(4m47ZWQrK<>5i(j z7m8s%gl8&F(GUtbHj*dIokTErdSvEED9}?8#bX?$NCdR|Mh_--+T0J*LiY{57MH6~ z?*T!G=A-W+x5$ZLY2d{fd?fPLIL;+zD4m4`x{@g|LjhQxCrI@b3!~6SO=BI2hJKXl z#Ym!NaF5dGO2wRU_#Qt~zJ5s%2h8_;;-<-zWuAUTht>=e>1SMzgGeHz4UwmK#1Q~@ zfix%dV*DpU;-izCJc=-(#FA%3n;{+$$SeeMicsndC5uz$sUXaA{T`PH7K$9RQl=1F z!Stctg&+lbKNKs7>l!If!pw_3<%KLq+KCuX+To&N435X+ zJSbrPoD;(X5ppEb1x63k`=pWRWboWc8tG-$lo$mG<{>c7JdCKkK>7p^l!Z|&im=E+ z;wn{&Kma{0Dsl;*n53|I-sS&DUkgUuNs>yv=|+0KO7)UWf-D2hLT{H&(!}R65e%I} zznFvn($Qn!N1o39y&}aV91&d*a`F%rkFyNHkn{_k`yj_dC{6^+F<)jp(tG8l z7ex_b81!T1>g{8DeqC8JMgs(_;GW21EHcdUG?jc-JMg-%DuTRJzS?79DDDh#u zFe-FH=B31H7D{HR#2^xwsC=2Bzana9A`3AqE8=h}iZ~Bg5=kTzg(nJBPkKoHlNkAS zK+`M>5Df>~+A7B^k|zaHJE)7^!D5j|L@);b->p6)-bmsw(@rT=0o+@F35x~>B+etu zB58_BdOqS&U+-<^NXlYH>$F*vhe^y3{b&~p*Q)omg$!mx{GJkB5aJ2U=_y7NLOsFZ z5)p(3u@_>}LTpNlHM)h2AX3PBcz{V%`KTNS61}BO5Z7tpWFsYfeRe8FNQzWluRChz z0k;_u6aEa<351_{i9&*p3g+-Zk@WOQ65k{4*T_=+h7N-qF(dp3T!-%$k;r9&3Ma(X z1I$KQz>5^E>cbl<1d%YjLOWtOWYiJB^xi*@XZj!uLKx{W!4SMhdTF~LzFf!{x_(k* zn2CAl<$k2ouWlDZ2!Ba@?}Vn~UW8}>u`!%St}xVLHJQM5#6EE`F4N0Xj3pxSBR`R8 zhzsf;ir2@$cKLTW&Av;kpfXIl|ltqX^>4jXcAV+bkkMigj{TVzGb$4`k zx=C>aE-+BcW#p*{PB+bAN8~NEm^MQa9p}jM5$p*JR1cw~ei6fX=p&gTYE~dxL^`fw zga~lpNsa`MBd+yhiD@OGf(h+d6Xb~qrSwh&_>q_giX&$yzOBwZjLw)x0g$+g@{C1E z79$kZhxSt9B#hA*AVN>!G7y*P6)P4njDt*XWi2w=?35#wM&J_bQ)^6y-A=3Wr` zz7z$=KjAP7o@EK~fJTl)B+#A06G5B=n4zL5OXI9SXG3sH)GRL$IeAf`HyXtW%Y40q z)knOL=_5VF)Dzf|xRI;!Pq zfxcU08QO-{Y0;cr=$XSUrXuD(+^*0E+(m@I*-wdJVNwKHAdm_~x!ypJ$qB)p-bt*F zqtL*VpCTC0J4BUUz2q_u3#s4m=xCJqt`1;aM5%5FXrots74l(y(9TQ3LeVA^RHF@a-@>+R$+(eKk`nD|66ecm|1fKMS0Y4CM)MjRAtNu4q-e7vG4VXTjml?0jtlB( z53Wz4XUGLTy{nUF`rS^bcc)0b|A3exa(J&W&I5h84w+^y110p<0i_Ty#2HbuD91FE zWQeZy3UY*K13|KnSt?@)Z6hL>K;WLrLS-0Do{s+YdwzY?I`?@&)C_YQ!i)&9wXgJx zKlF%P!Mg=2)Zrry3;_xTXCfn{p8BGV1i2Wxhu%Ld1@T4(p6cVz9%5R(52{Fgy}}_8 z3&p8|(P%L(M}WrlVk*)eL@FYSJkQe$kpVogAf8;{dV8*Zorc_6Z##<$rIppMV)dpm zT6mz3*DJ)th%6G31QFbBs3HZO^a@46iD3FQN~rUtQ13ksGW`-zU&F$=UUQ;>6?&^! zkc%{dLzMbGq~LLs=UnL7lh!o^lKfmG97-^Sfu&!Og;ME-O-xiVtu=)ohCYI^Om%UJ zAEbzq_1jn;>*EZJ2$ty>h^JnT=_Kb7nlokk^iHgg$Z;ZA&U`Kty;Bc?24WcWlq?I7 zRpgQrm!pNxLoW;UTk}A_hYyfx>RkvD6MBG9k_Hw8EWz{(FPkF_Vk$;LfUGqXTp~Y6 zh+qOb==DMLv=}v0zKG!83f~WuB)T9yqahZ6mEb!NNhz*3_$C=fr_2)KVNk?9dJ^vz z_%H{AG6^Gi>l{v7w=``=7kcd>gBL3>2z=dVQ-o?lH%g+?SP;P&(g)ZG#!BH4bzY2w z5iJYnlu}1P;$wL0&D$Y^|B-rU6%sLh))(PpPSgy}J`hR@%u2T_nDLZvE*WTf9|(ZKXJs)VDe zF^>^(>m7iyi?%bRUo+6a^pzpVY%uroB8oK&RW0wA9YSOM$heu4g*_&#z*pPC7aqKH&~#vn#g=<8Ju-TQSc2`5Q& zK{7ct5X1HG%35gS0Jg+d$~CkY}e7 zArePKAOURzNEvzxDF8Ycj2Gzx$C-Yoi_Vg%ROCc3k)hpU(e1&{W5&HgvMhvtA>v7z zKhxV7^c5tzww*xw3D=+L6-OE9Yx0P#sSxZT@+o9@6kT8F#eHP$3FI*v3L==!yU>`5 zC;GjDzB|zmn3A%m=k;HX08wkYM9-0;;=)FZqdnb(UpXA7fMUe?T1^6*% zL9XY8P;XE1h+v2bWK`r?sPs*08M;1_lSIE9Cva)yf5>B)3mwW9slG`n zCBBfsOagBk@-WMULbx6WUJ@gm^|LfbK;#p_WLyYLO^KdJLV-v=f!F0Jj~GG(+KjG3 zFF^{x5E>V08gY0Lu16>A&%K;lO{6r4f6(D05fveFsZ?)V)oV?7Lfo;W5|8U6Het?l z5qn(kLqJ?xup;sGF)<=orf(QSU*jx~{n+yZ8X#nUUIm@&gZ-~W-zB22SDcaAv8X=_1B#EBI8)|OZ=H1aHq3yv0!^a)2sTs=_I z&!WuNXED5}fCtY)j_?-oq`szyc!V6$ESH78O%su5!n}ZC82F$NsHmb2Cxs{s^4!xG zfOz5Y>y9EMLm+^!U26@cwZh`W!_;UifH4A|3`PBJ6t-t^lwwA}2#rOq1k!dYScYgI)$8H~2LeJN zuLu=-h+c9HiD1ZEQ@B=r7*Q{l01+mO1adsMUfLo_q3?MEpvbtFNI0aph%sHINZ7$c zjDnC&2+z}ruReOnSQ6@umrBWfdp<$kC)AMjkC@zvKuWGO=k5@0q-6RKuKpfDrp z%`H5JLqrrK(@5`aP82!|4NPycMw5DaOIy zcfKNE*Jq*=4|8RddO{#0;8GIz9uzuJjRGc91XLc-B)pHF=F2JnssH)uN__Y&m2KTLYTPp2(~EZa z>*{kO=dYy<^*Mhp1u`RFOu73O&dqm;?!JbjzsmMhR!_R@;KMk(S{l9_vj6;aesJoh zb@x5c{u?5@?vG9Xu{d+ZPsClpr@r7XFX{~QNBzS$yM4K7_DYmRDH?MpfvH}g@MiGU z9LPSs$^Q7$U%vhM{MR>CHL28p{`T+cct9unyJ|nyhYwod?Gx-{4S6ZhgH+Yio5)*> zR>$4`v#BcGgSW3=Rj(K!x<9q`w|_tGe)`-Ew*S$-pppOmz;>!vZ1CNx%jdc-UFzzi zRo375dXv3;y6dXJXL`a<9olV&=ou`&;?)Zr_rVFS@drfB!6VBdae(jX=G<96Y2TcG&$pqH6_a45#g5^_)~&r1ux}2yp*%> zyB&pKw;NLIb~7Zm^Wj|Cdjianb0PwhfueA90z;zi=NY!jzKfykJx4Xyoer#eCRC*> zeoxZpkmlLfbPvpT92s?cI;_Jfx1NiyL_6JKpo29on+J z`%0m;(2(=;XqLG-e?`!ev^jqp@Z0-e=cvwKskI|If4TPC+utlmzWaqp^NWe-yI*Xm zNKwLB!bmqCG4#bKbM9HfOgKCL_Qfh{s~Kj;^Y2!K)Y1JWWTv$m>F1K9@%|<>$~-^z zyOIqlp1u|`GdJ_m^{He_Wo^xjHp3tPto}*l(;`Mj{lJd1AyYOZj8;Qb=%1+_xEC_h zc8_eC$sM>C8fCWYz$m#t)RIoErN+4RGfQ$jp>yNO7&RW*Fw2rX6M40WkySsi<*X~V zb>P)vqulzT6~BY4lB1qV+Bz!rX-_1Dgpm@z+YlC8*0zc|nod$i;?px5@^VM+SX#tL zs~=c#wu(B4w1}A#4{SM`@;BwctHnmS^+Ox}ZA~>ZB#f;1-ICDv=#;)UM(TUEw0j5I z{V46;k~W~T24kdwC1>xz=^o=+bWDd!0ZUz66`9t;UIlt6G8Tte1|d7es7_Ohm?;&r za+ma|XDsBgjCXeRt74x@YGip9GgF@)+VEHWbE~NAy^xW+e`LihOX@74Ma-;tV8iJU zT8_q*wn61VkwYvkV&v2h>^MaU{e%PaS#X5^{Mb&5=7o_%ypz-jfA`FaygpXcw(exi z#QKR9ZS(7anhAF@W?KEkhStT<<7T!tazl2P&4{~f?A)b8h&LME*@kw_ep^zTw39Kj z>L*sT^o}&zkF2}VuSa$u8@Ll1;nj~UnU(s@fm%zAaqDMRuuRkyK^sO`8BbZ{^JV=chI*ys`t4 z91=!a{BA{9s*hAgh?ptyz=Ct5Fw}|BMlWMx?VrrhHXFS$Nh-qFy@=eCULMSGey+sk zva!yN^DHvpJU^J>Y`2Ov?Zzk6^CEK3d3j*R8PGX{G0wn}Go*8dW1Qg(CokKNWurEX z5Gr2D+_7Fgv7;^ZUbL3X+>4En_Ydv(>jRm)K8*0z2RqJB4y2z(SwGFvZfo&jvt!K% zB{zhfOm}kk^tVrER5;X33f&z$-)27t-wWO=arf9(i?(*qpsh#Q+j^Q=sM6kD>cV$S z`llI&t=>T_n0T?D<*#J!`QDz)(UyAxU%4L;l`|`EqTcPxEum6<`Fo!J*Ey<6uvS(0 zAA#;7GoZXYnWgLmEtJ(%TuB?Jh4 z@gdjiTwtKf1T>zmNXgfyPe#1mTZY@Su64up{ITFAh8eq_OXNh64DcP(hr;Oa3& zi@;TjU|NfSq($heML4a6O<$;9M!K!hPdT7ikZ(M1H$uz8BWRuD#O;P`At*g?NTEfrR=sy$+FR& zZ$r%h?&JpCH&16(xE%cZvp$ncQQ}r|sK)JcdqvoGyFm!959_&T^Cmu_(pw>#UKCeq zke$(J*2!KAT#hDFb-A2d{&w^+K?|4YU<_bNE9+R{TI!(0tCLn*a6%jXn5M;MTYgZ^ zs&vvb`gCRNm1-1WLG)g5LKXdZPK(!4Qf#4%4;b~2vpl<+Qb%2U$g`V^uvA%5+(KXc zj;H_8hG(;f+ai*6GDgPk$qa4fG)vuz-IG`La}@2Zlj(hv#hWH|DC{W?#XZ5Hn1wH^ zR|*4)4@vOj3=ieCmEnce{HrR;{>0&)#Mym_D8MSYk>#$T(41w4bmu3tv^$?dxbufp zJAam>KD6`<=}vA)eErl)g|^RHt1a>jHLD6rvEj9ZI#p`Z=)B2j-uUkr$$KbWfsY#;8gh; z#fw(|&tAR0{Jj$WuU1q6l`All1y*bom8~+f*_p9U+Ij$j#2(P z%c&4oA+LK?o1NNGZZ;c8-fM9Gd|r`myOfI_*MbH$t{=~9u~+PaixO9I1|6=R&Zux; z_DhPo@I4uRnqyFmk|m$Y4|uv?t*D3_yKkh-v^JZ&UzlY0MfSTl3QE#86unbpSYTQg z_%oy~t$mL=3*7OZf1Y8Z9N0G;)t^74U+&zNn(pKVyo9BuvZ@H3xGQ~6r+=FvX=u0p zoyy-6H5PaNl}(}Fj8V9HfdN_n^BfzZ6#^AX4|wp~97#nue%+NGkW_Pru2Doi(<;*o z45_-G=h*f&(R97QfUW;|j;%Rz+8rlvA_Knw1nDeZkFG@^ihWW}t?XDTm5%&d4|%lYC!`ZCJ;VoUqV zwv7X`78@nk4=MaFjtaa?YG7vqhu%Kg$!+`g_DH4N9?gu~qb1>{K&D7ot+*E)qdz^i z(t`9-=Ss%RJ^w=IRB-Rm#QV`U-rF|vuViEYI%4Ht?abSdcEw){nQ5E(-SO8Yk*Afc zdOmL1!AXfL`3ZflO4+IQD#_Zs$`J)#r4=VsbeTj?VrK5s!&&}j&jhIxbOJ-p?&le{ zny`$b3*WQUbIT}{+uDZrw9VeoM!TEMbSs<1eQdOw*i3h@S=_!xx_8ZRJ5lB7_7(UIympj5X)l>N)jsA6p zigK%;;Xc3loBVdFd3ipRlEBgn44C?#=h#~0==}+x4e34PKytu;_uNJir!DtGQKKsM z+wA-DP0J4Y4bHwj9;nLP$Qre|DPgBhy?55;O>{(=w*oV|5RMQI&kxQX9z+4!NbbuI zX)H!E_3FxzODpZJtK8{tvF*3^3t4A@d$!Zh(`*&1^tRQCV97U;JI1#MGo0#~>36R7 zj3N_ zxGaw8)JMwu->&pM(Q)?O@1Br=?FaeSelGo@f23yY3pJLXr`gvZ2}bKm-xKNIW=JT9 zwhbq?=Z|aekL`wS>*{$!^1HWU_pCP$r}@dpdPffR+8yaVmy#P|lzlHa!hU`{t;K=K zl2!J`K9#@YJpDRHRmpny9ao2{K5n-qf%8^yKz{q!N(<3{+_0qFAv_c{YjRb7f;WqX<`UUdfo5 z^%FZ<(zVRjLLv3_c_A7B78}$FFv5sKhE(qj@xAik^}aJ zvSpU~jR5o40j$0Ya44T&sGEzfWbP^7p3KnJgx&7C@I6sIx7(fesRPqT4lF)zK>M(P z>5~Q)A2VneVfjNBzN64T%`Hmp#ss+#)H(0H!|P4rif=e78a zME#qA?>7sQ?%oEB$A%CT+zXnhuO826aXB@xEq84N0p1zqKprjT>j8cmE0lnX3bQ-GJ|&#PJMZdpvFja}<(+Vw^|n>Hs|*L`R1fXa@|j zg9)xf4XXnO)Io*PVST=T#S;l@mqw(;NC=Il7o#T{okuR?ic@%HL6}gnP#&d`7bB;A zoxv|M<|{mW>AQSSDBlQhxyIxe8=*+v+G+IOGjd>~Ku9B8Fe4GN<O5a~v!ff(iPtO?GTL8<;uO@zi6+IUa!X+FMN|Oj9mtv6k+Hn>m1VB<>;cT8uS=oq3B=8xLQ?+DE=#-po%gK6o$cPh#Vjp8tR1_6T;z|(w7O}@@kN=nKIHbeH-Rb+nF)`jzHRZ zOp%#mQp`6X+FLx%+uJw%wpwoM`u`T$O8Zvl1pnOFCE!WR4Bt-$q0cR?9S)hcO!BMf z2$3D6wL>D)5G1s1M>7PXTi$roH1G4bqRWrP)RjdQ5yH*vQ0|fh%I*`x+#-zbJmEI< zhQb3`oXAF^J@;kJrFBdOhLXnU>k5P5SxL|&sA%9F1jd{4qR0qswU3%L_<0y|WW!)1 zK+>fq!Zypkm;DKSopJ~p5;0p~CZ)r{}WV=;)h zWRFGvz3pDvi~&DEH}~c5CFm_|Zhfd^N{r;~_|&!)7_Te_vB`8Qj&MUK+<=4|B;l4$ zxCIHfF~Z;Tet&v|KlCw(ML4BK2tVB3f5vkOYl}uC#27;12p8>b7AqRz0&YX(+4wvg z49}Lxv-Nql7@n8Kn1DE-$Nb_41jKmSq)ONu-+ZigOEJpwGs zF%3sWsp<&=(QS}r+tLZQAYrsJNzIMh+Qb{1+xkm!IF!wiK%yZwl}?Qi-jND)=a%R$ zUZgt`f$Kl+{+Z(b*>V3IbuU7C(LQJs#HP}z5yC(6?yx&`W%G(>oG}*VoQ&Xnf!7ow z2#@c@SXx-XvhO7$(n*NNCsQMYe>SYon|(q18BLT$Yi2~4?{_Z`ct&AuF^Pm2gNR;1 zE7%0FxpZobu*hG3J{OdidV<)5IyFY9#yfa6X>U`>q!>?o+y%EXMKLztl#JjE?r4c( zJWhLCO920n|DcX~CWys2rQ!%TbixftxIq$b>4aO5Fxn}2AT?$WZael6Z^9nh!%!9P z^D}&E9D%YKjSK>_IQm`7|5=&{!$DAwVS|&lKb6&?D;lyS4TPlVJe-6;II=hhqBdib zEqnM{R^kq7F^vY#D^uW*6x#wRstfGW!oWZgow;Fa*-KUAUy~nE6ATG4^O17e{E&SY zH%;~lattsi$H>RYNv)WLIw4g@lh@*;a-s?g4Qa6qaEi`($3grVr^}h4kd`?YCqy2B zNPSXFFeJvzN6Hbq;RE+QBaxsGorzS@!yAkdD3HB3_7^Yv=)1h6e^4X}VwM>MW*Ibe z#j^=H3F>m}2r^0Y6b>*bCq8E}S`SU)r$z%I2|5ocA+!Y^0|V);J@6PW$<3(^4~xMz zsN(4RA4Pt+SqRLB#@cec?1+{p+W7Rz8=F2M#-)Gjt2Ywn?0l?6?`LDRPzTEn5LSU2 zK0iO~XwwiQz@i+}VpNZh1e(W3o8s{?=AQKMc+U>v(Geya8pX!r*dR!*X(ZPk$#smR zeYz!ZaBG`_Cqx)_qzUSsJ3+mRPf+iQe+SX5P=;_b#%7z6W1JBqG=tae-pSc#I6QKp_p`!=Dj^&u5$g4PpC&FqtB(Z5EyoVW2Ixcnd9v zMV1#G53C^c1`LqRH>XEvKc87%pDRL_SR*V#ofu&g@ulAHTe%)XSl_l_7O$0qAwO1a z*Y@GNbI@2{hLugw623ME4-Tqv?$25Bcq-DW5zzP{zxSs;hu&uohC)>xLvEq@ZpOxGV#h1e;N0OA^a{U5d#$4D_ewH5{z`3lpfW= zSpfSo`f#TQ_JiDGvH17w8xBLhl7i7TgjYGlY$+D zEy8^%i^DR=7h?r`|D|k1wv2q2@vm#wo`0rE3%YZp2JNWQ_absPHn8uolaYCR*W;#1iii?vc`Cy>TDmn?4h zQgo}uO-?{@Q!Z89Yn2_!x3%gFFOWPVo2sw+-&$>>2T0eLMb^X{U)~;^Rx308fci`W zs!E=DESs#$n^*qmuGI>7gA@a1WZi7_o2*@c3R&7){2#Q1OHE*E%g5@r?fC<;`Zci~ zK@x2Xs?feG&sEW{m1(+xM4I_%iFIPXxpuqEHKn(FPVFe4y*~1p)ICn+^=Qegn^fA3 z$GagBU#AjZd&JjhVwPCX%&liespk~~A861+9q_O+vCySk=#oj_v`8H2r~&&Du#wT#79%K%TPl`I%%bBEWoe01vMk2mk*O!-Uz z*!h$_o~w5c?Q;_`r3{1B?7qPL$^RUvm5A)L^7*vi*SSY!kB1H8m#C0Zrr@n+vH< zL@ixZw$>%6IcS|+kieHW$xlSGFiF-KI&zTaD5BIlkU(Y#Zaz=+6cKB0O-FJ(KkUw3 zdrlWQIsmnec|=LOyN~H|I2e?|(}^<16_i7<2QTwV@B&|tXBZF_ii@xswVQRVOwSdl z)w3cB{??XF#`dQobRZ(h4=B|(A?mE;e)G z*KLy_$0ok{c)^`aULLfxxDHBd_dk~U8oJ73qGvnhb2t63ovPqZOT*NsAwZ8jJtL?f$0T{rBB!!{X}9Fz-%NdN;&{`;WGpFt zdAd`$F>5T5MqFvcPN#?$gr}(c5uGQ;EsXQ*@DdnLkJznOiZi%#96m>qZJ}I@X;&la zv-)?um+-s`2EUv^BYQr} zs&5Gqu^4|iXD#d;J+n31jgRfZ3+B5DzlWFiI?JlRr>^D0hI*#?^)nM!8w_5gS66= zcgLY0*T_`P{}u_CJ86A4l!xLX)Ry?p0`1L0;{v@mzF*+If?uS8$tKson-$0&-7wku zjn}bCGw)Gx-g2+0mdRHpD)Opj>UoP1Eqommr82OXqmQdwL2eV~Rna?{bfqy)g-u^k ezbUo8qp~TTzNFI?@(z2bec6<)`ZZ>%-oUC#N5i5@aJo2Yi;LX?C5B0MCj~j zZevR5VP$SW5AC+T*=1R~|0jX$h0j-n*2G1?YTxAc6LJ!ERy$c)v26TobKL5|N2Blb zYSB0ZKn*w(1sK%#c*iCt)Bw1}IWXKw9QS_%>Aej39(Qzm^|J%O_uJ zJsuNO2!DP$Htu|;P7(ONos!+Ct9;+v67k@#VQ{>(aY?j%Kkw%rNcJ6elwRLFetNou z-lOm|bn&Tmp{5~et&ET_}CO+jAfp^tn7snG>FKh3Q~#c z{mEX->Z0v~y)+5$rd=D*ww28cG$dEztL@)>4mx>EUi&|zl6eah<{m}Kn* z32U`zTKScL?2+p7G)V)x+O2}K>inGsX!=_HdeY?vy+kuO9`;BTbei7Ry}AHxojGS# z?ar_xXl;D-mSVU(f2y%MN32?S$)^o;JMi?9K>F$f2@U$1BWe$NE&qt2+Xufa)#mij zQUw_>HEUoYNLzSy2Dn$Nxc!$bC;X^5Q)sg;S~KmWki)2?M!7);_Q0yt5LP5{z1hCDbHI@6-8;5%5lvV;sP>k|cJI&rX>Wzbqx5Q-M6#66N0DwI*vdu%3M#oSZ2 zKFB5=RHqKp+^O{>i8f%R7D5#!c+pBfQTOIhE#axtPL!iNs=tfw_5pbKK{-ktI^EP& z`T;}tHfQAJNi6Oy{u&C^;k*hPae_LRL@RQR+}DdGHer@a_)`}*gbzjQk9W%Ya459d z&mcVPA{%K(nkna7&|2}TAUVS1tgt*w1Qd*Vgsj4r2?$g7>nR>t204W~JSa5D0PRPU ze6i0P_EV=711R=?e>qOgPgJ}^!6cu^X&)_UJ0g5nFVsH;BckV$sccpg+q%vQ zdLa{(Ajqa%R+t7qM%>}34&R#DrhEi4S_I0>{dS_q&vG-j96#R`{&KVRFa*c;kihtM zWZ=_~|GjJ2U?uMu+A3o(y2yDx*zRE%A1G(ym=2h=Unvk@J25p@F3Z7TRi zl>JoUEgG|=#z+kwv(U?%m*%88W9fS^v3oSI>)6vv5-;lDK@_;)7WKg=B1<;{yOZ|4 zzu2IWOVU;Jf1`<4B{Pkty%i|LJR|Q?Pi?0aWSa7Gz>Nt{M4BdfRnM^IcT9rQh)w89 zM_QSluPv=sKa{KuFZDQ)#1@Jb2dha1w~4qc=f-M__(4GV7IpL-?r-45+X^MOh{ z(J)<46h0WmB^tjT%<4Q*?eQ;M$$SB_buyoC9aJ9n)n)GHEUTdgK9y48E_yK>r7B5# zJt5RX7wv@PsYUA!wbcDY+Ljrblwi|M{7xXN^JP#772IzPR8t1ihZ%b?TFvQ*H?^jE zwC4QQVYK9W^H!RQ`zh1}5#721cJ!`ciS?J^&&XO;LMxfjY|xQl$GcE_Hd@d0rRIhc zOO=)I?ft5n1nHi}c7+|adi4jv%5$SRlEd%!Y_J(bk2h{%_ITe2s+6lL>}VDMwcf(i)lQJJ9lh zuK;W>k+WKb#X)#yK@)@2M#M(bg+N|T6kARv--&0pnR_-TpQ@2JY;1wW1#h_E)Kj9K zkfR>qWM5U~*ZQ0fOS6wv$eU&CcXpByVj7AgNh*%c*jPec#pe;xgxc~~VUqC`C9m}O zL(8LG*Q&Sch6(}pJum4g0vz{kh5G@F!Ug(29Q9qVH^ELG#`sk_(33_Z%>ti7(3eOU zD0LoOb*7E;nO-o#FGi7AT3wi%=C*FtzNvSQ3Hw&4(GI3qP{|I_>|lQ>U$2xf7#z?Z z3ccb-Jn$|KFZEq+JyYjFqcyOiwgar(?$!|>@7i4iigI6Wf3Xwgb4D~8?`Z2c(QW=MJmhpvbVihu8 z{BOBVgIFA71W$<}^BxQGAk?7;tXjpJRawn^y@GweyWc};-HE8q1bY8;eD^1u?}sN# ze*xS_VbvKRUtjfNS#qJ~CW&8(w)jjEi(98=sMFUvL<37CIJXX-=*{a`L8?+gK8T~=IvE<=q&yzx zNPvd}sd?!IlA-p56&pqC`!W}9K#;7dvdpWWSet_N&D20nH#=Fc)~gSoC(pez>u1_Af1{mJVJY@{sP{(Sl9v29O-w28O>zot-yaTr66JmL z;4RIbTY9F}cJ=+*liSB@$J*7+Cg7gTq%hplO&QKyy!(9D)z}riLNI%FCbYr=5LT{{-~E=BR2iQIHyEuFob(f#<5IVp7i zKJ7P!{%p&LgRXtTk&dwb^kvHkGV(0~z2{oPcgs20ng&+s|CBj7{&*I><4n*0S=0Re z@xFlX%OUEhcFR@Xr?9|9X56+8%&Upgse|BU*v_>5cuiUcn=~{hmfpZ&Sg~U0VG?k; zChydWdAKx3c_bcUz|qRv(=x8rk?lk&Mb&cHc3wMH{SnTw#}GuDS-{lBMcr0Jm(q`= zUG=w!kiI-2PLlg<-VI!18OBnbq5N?B;-FH3V{WT#CMmF1^gmwD*!%!J$XY0HyIERn ztqlM=y$oEF3Q_Y~b6pnehH+fvEl1Q~uNARl9KNH}kxbQ)e5+Zir`X_NoU{K{QnAuM z6WPG%@xb3!lwI|pjXX}}jBM3KYbAdqx|J)z2?02Jn*4PR2J~L7hYAUG>}RIV23(a2 znxxqnAWd%$w`-=H@k^5d#~f710o^>m8fWyyY@AEt8Vl{xkEDbk;jB z)^}yLWq)*x=Mc3FtQ`|yO>$e!A@Kx>@nDNtKj?MLo5$*xKVyet*Mw-%0a^Acn0A+z z=_g;dmW;W8<2-BTjA-Czv^faUnTKY=@6er>GgUKNJv8q@cIZN{;eKwyc5zliZ)Z;T zy^Xi3Fj3ZAVY2t!2y*<5+PZ>Rr-OQCMot*-@whsEJ0lH<}Z*t_?rRjeP6K1QJFs-Gb zC`TVrZ$h-gU@a@sX|ajb~C2p%?&rmL|iqF@Z&;=#g#V)DNR&@8< zgGH&*fiQ2yG9M?qK@8SGTBeTuboUbkcMZpG@!TFW_V+2;wsVVjSRm zWV0IqDDmP0alvD>(S;%l;I%0P3K5o*jG%xIzCv1)o;?DkN|Dy6-lfv$*`c5|^C7w6 zJ(KY{W^~4OrhpwKr7bk>Z^HavngnkeOLY303N(rIOIuXuK4YdNagM)R-R}IqMHWTis7nRHp@pH{c2!7zP%9-2&Y879I zQxbJ3D&?uxxR#o(=8g0gGjHymv{EW)36dI@cx-E%j>Z_~O(jkKHEDBv* zw;QI-{EYn?D}_y>vF7g_vP=xX28$n~WfOU*B#cF7Hq$yOdLb^209|zJ{+J@hW%B2r z0$}XHXqNv(LR#q;vr(wiKviYAWIWYYh8I1sN?$JCxm-*dT<>&aC4J)ocXK8J&mCM3 z^qA#xC-EjpJpKwtTo>o8m{hgoL>A$}wtMDu;GqAoP(uaJQ#i1f+C;9rpOvJPz<#sV ze)EWZLq@|P-ApBGg0b2~-lKQqd#+1y!x8unrepudQ*=3xBFkP|-2Y#C2giz_hd>s=IIAf5-OV0JaK}yaR2Ku}yZ<=vZ?7!x zOG_3~7tg&)CEM)TSmUILXP4FXBl=m=%m)8Eaw6rEHYh1&J1DBj)V}RAf|RWdr|>wk z#2()j0j)^|;~ymvrg`7_>R8K^IXo!9!?I3G93NM+{|=Se8+h1>VN<3o24`rZh63C3X5NpF>oyaYZ85IwMW}Ll0jptP-ZV<3`uL%t&qJ-2Yrqu7gz-N^h_Sqwae*SjqmY=Iu$cQ+F;Wg{ z1Gs9DIIoEI}N(fs0*Ysm|40HaHyOXw58T(rJz!1N=OgvO)D zcPmY91x8+mBuO7j8JBvpGiexMzTcUAJJn{Hlld{_Nk5?8l052l$877uDdeK=0jtot z(a~c~kFRPa>t7=nOWJ>yBOMldY03lH-{69)$O?;e;PrEBF^L`xpesjC$itP6VGx)- zort&kXFE*VMTP~ZZiFqf*{8(_zme=djV@v2+eU5fa3s7_Ua-lX891JR;$En7m&LsS z1ZNDVI$*}a1mx+jN#pWN%B10c28rCg2uz;6x{RX{8kjj61=MF_zkNge|MqIaXx5H) zRR%6a3DG-6>UWC+Gs9p7U-tiif#x~u{{;hiEIHp#5Dv(zdYU0JjwJjPAkWca1N&_+ zVdB>y;`gN+NDFs8(g{XrWDZZM2|h|Gnd-AK9Go`!n8gX1&X+ zWh=tYhb!4pz%OYoo1Q;wd_pg7SoD2gYU&ymziAc6vmq=G@7q@SZEtF&f7E?XJA233 zLTlkWfB7X|sUxoX+ai!%^@D=Pfc>R~C9K`m=Dx?8Jv6QO0qt z{820tgYwa`#w@LNPjm2jt;JfcK*9=5>S;X{I7Vy~u22Vg_dIPZCzLo>gqY$J(*c*+ zI0=~11H#c%j#bh~ooVm+(xT4h`VvKBTO@nFcyGQFBI)Q=6(hSulw z*2`pvJC|FS)7>=@<=evw1m@Q9B)vq~l?=qzf*wNirdd!e1x3a!h)c0%2c zpwMJ~4epOxtEV>AZtrUSa*p?#QD!|_ao`hb`VqGv=Jbz5+~ZYS!T?*tE?L`NV)kJK z$2NK0GLnwZZ&!>}ZHAAM%EB|sN~r3qXlKiA`Kc4QE9=L?rrFk|3mm%HAJx#h4x2$j zb1tmxTS)eHx}kJ_-M--nbqct!ml15+$^50k07V^L0}rRvj)f^0?9MrX{uoj|CWnWq!JOPD@p z`6Qp}K+W7Rj16I#3&-pexTG``Do3FYHOCLu`fl45K!gv-3D6PV3P!`z!R2PRa298< zD}0X6S%?=^r*`ThW<4ZUc6>2--?TbACgwQB)p{oH5=VH-$Y8$U>;JF&$Chknajc8G z0=H>$Bk(()v0#AJr44GuoR)Z{8%8DN>OC6zq^!!v!Y@`|r;gmOwhAq`5LD0es)Dpm zhg{C54MwJkpnC|qJa^7Gz)3#VNU6cFNw8FHKdc<;71XG);DxoeNeL}quA8M!v(-zC zKXag#X8#Vd?iUxE1vgV78wY14S*ik}P(n?>@NB8XfT&5GeovrYCWwue7px3YtSG-O zGGGF!!T_n_vYeC&89pvb8u}KSWviu)wZ6hKEt#ZUJ-#MNDZUfG!BJHzDBBs_9{l9i zO|>N2@0Pad@!^++=*!@s-Uw|&AG;5M(q~0_`$iAz}0$ws=828w$nm+!LwB> z)1J4r!}+}D@I^T1&{q7=6M5hP)Gy*FdwRD6$Sv?W z2hM-$-9f)hSoU?q@ltFneHw&W(<54kpd)JaRA)cS+up7+>_)=OIm^_~a}J1QGPWI7 z>cLZdUa1Hga?wTWH)9b?b+XtIPYq~t?8j}=L0=?={x6`%hmA=7dT?c$vzQq~t z)RyP4fYc1W40$wZ$|BFdcqYLZBHpEu=T%}|u1XOBggK<_R<2YbKfWm!d{Ha!Sx5f) z40W2BB(D0&L|3ZB7(=QQxi=-OvTUHMJFB%0ru6E@9{#)IMi=T9$$f;J6HecBLh4js z^&*+AP1TJ&x5j*|whc+<6>RGShN(x9>ViJrVBT1N8|?{cUeWWmw9$-I3|Szv?uNhK zXw=-;^5WH5ys(_bPD~U%8G2;{S=&(arrWFiZL70Y*~zrgft8!6K!4vNri)&k=0`H+ zqM5l1l;SLBDka4I_a#jEPbdnNI%k>Z`(!PwHzfvwLv=Wxbx|wTfd|?Et+s8HOezHR z)}mJrm=sfhu*OoF>IAxt5a0L^vq{`z1DB3m*FX(pyHB39nO)_Wq2EOLU&47xRC7Fk zgPu_TR8@DdL8?DOC`FpR*j^IGdm+>o!D^sBLZ5d{J5AqJ7_|fl!m@g`CfAR&U7h|I z&F5kR@jgd2sj-9cYd^9A{%SCWJaPPoT+k!plpRMs=|mOA z14}G?2>i0J35EDmaw!-QL@=9!3W>rjS1X@qYmi`kImh_bH(IO=)%@~C+#b(7@ zvQ#VRa}8&8M|!gn*$YN9C_}zCuFD7B7cM08JQCnS{Wi-0PfP!m31E(kz_B3PAE5|4 z5Zg;vyluc~_&aF`Eno{jSX*Oj=$8B!)dqimQs6#)H1ss4K4~N<~ocW6stk5Kl0Dbx)v>2@lL&Z)U20wkx zTHd;!Dt7|9OihkTFWA?j+w57n6^bepEULj0jx+vJjM21FHvPWX>;thTe0f_!!?V(y zZt-|&wvxBjxG_K;A2es7L#G{RnPxRlhwR?uQ} z5<}#@6LGzK()FwlD3L7xA>-u?F$Vv>i@ewkI^`v4%#bHy=~g*`jZZkiYY``eJ+bFoh+^Vl=dG+NG3Cx@qSe{5G9Z4R8k47^Y6ubzAe*wC&-+PS2=GZ3 z!f^Dn|6A>fHSE3KxvxMKfwAPaHxpN?mA-Me4q&K6`XG-@9$W6NljgLCh{c;VJG4}k zUjg|u+l+Uy=hx_=@Zn~teJNP_xOD+th3!Sb$_OHHg9bdaUC++0= za(TJWxK`BwdA!W?svZ)sUo0v;n)jHFF!-L3T|<7KBObp`iAopPutVE10FhoGZIl>S zrAhFG7!)p}ZWE}Jv@YGkNOwdo;#O3*puA|m&XjAmma|(Y$Rj#ln(V5n(1c~0aHflp zuJ}U=0wW0_H0+@0KY8NK+K2_LT|N+cm7*V+{i%+EulW!XqIQ0S_>^udHeZx5%jc{$ zBioUP65?B^&w124LleYCCl$@lIaMR^D4#lKrTr5n{0e}f2R%m8>DmmRDYu*);vrC022bV?S^Zi7@P0)9of%?TN%j9t~;6Mg9TQI zV6ZTr9SQ@zI2t$SecotUHGY`yUB`d z#h7;dRLg8#jgq8>qGuSaVx?c?olzW{&uauw%de}Cq-U^1= z;;ZHw6|m(4mRV$sZ^g=vq^M1ygb8jOe~1(u;rp3Li)5P8WaB2KUAniqk+f|=_9kZ) zcocBo7!HA|8hiNo`2}nzW(V0)^PH5-_d2M@VOL4i1`1m9+i>pB6b+T) zFwvRW=jEfdzeKCP2e#+>2P}r0^+`-)t5=v>i~FfE@txEHnsmkVWHj4Nvg7dUs9H+T zlu(dn6g);TG<>(kqkg*^QQE?eQwYBr%4Js59>%jHZ;wa!>3m|ggV`FJ{_BF+?Q(bf z26~atRPDg9=(06#vKxG>sXb`|yN+M7@jJ$*9hRV7jbps+jk}l z1TK!YwOwX_R3m9*sI5c?rS)DNlyz3Ta3m_uxe|f-RddHx`J8=IT$CT=Eaqh4X@yAxLM;58p z_!#bgKMg459ZGj%7`kO89UT8 z_Fpx?G%g7}i+6~5kUOfPuv#IczT+*e~KwFQQ|H!7>sDzBRV zm;&~@TNXZvIia=iNb#DuqtFef)?}cSU)ibUcy3J=uDA-=k!b4nYmlK1CeF^qMvAcn2am{22sxuIS=`-NbwsrAuRuou zS)bHTR%gLK0eDV|=PmWx>vH}YxS?S~e3%8l$RkiO0Rtb2*gQG-^52_9ZDQaPnW>)9 z-r5?A?II+2*8zHvmi!5@sA`l?n;8?3@D3VBIWfI0N3!$4U(%t#q2}l^4gfoLl-2pa zEdxuJ;v`Oq3j;S6OytB`k}*8b>nbTKKo}%7##({7^D?|F9%u`|CYE}C1I~?_>@eBP z)DUNZlbagjs{Qc)*o;XyGi&@67MU@mC=9NYU3*jtoC2b}#kHJcH;~({y`iZIM1yt8%#h{V&YBnvRYbjN)^z?~s-HK*r2UZJ?<$k<=g0F3mp2_1RPK3o78@|8q-TmY-!;JabjpQ+3F`1UTt&lmYJxM3H$q!=8%sDT4 zR%q!dauY-ReQ$3v`}*m=|y%32_yyWExHW8bM=a59d}h7fS`gL zFF^;3WDI(c9#)`fQ02NP(S}nVdiFeZN*LA(yE%TUSJ_E4J)0xaq}N36eb8^xaLsY_ zARn4mk96d5p3F5n8m-FVs*O}9dev5eGb|97m*j^Yw(QruV$`fyVezg*WNts4orj*b zc^>oBR4Ub_C8KyvTNqqeUYvc>*#$L;Y5sxE=v=*UUr!z0$jf1XJsUD-f};F}ggH@O zol}P1Sp<~g5obyqQL}Qr$-YNbUl|7<=cdvymm3F=Byq;1?t(x1leK6L_VBy*c^s)% z^g2ng);%!~nr}2FtYKjwD~z!n?su*1EplDW4;(RCc$!s{t;;eSugRKc#jtbeuX?op znMM`#bYCB;&RcLMod14`3JG|?PP@Y62bQ7913JqwBl*<~Z20>?xj?ntbF|^(OZQvn zLo=VtQPHA~ek_!bVI#M~Vnw^b2+|1&3*`PpZ<{hr7UDis?|T^g>m&AlF%N51+345P z2*KAE!4W6*BjTP1TN+W=kwM-43_jBE!oUjvQy1ZW;^m<@BAMZoo6y;Is zHGcZ&N)E1)rLs0|Nx%_)f88?(WC)ip3PU=p14|cIjv`7mAn1NkQ7zq5GlSAdti48s z{IPx`l4|~ZbtaxiJfkiQ_RPj$30;tm`nM2!!Ln4@_G7BM6QeEaWNDyz`!0dF^@*7F z{cAmPY4&2^#fB{j@lNA_JBQ|X)Dz}s^eSn|dj7KfO3Dofq@mPD+iG~>*JYl0NaO+L z-QgJ__Sp?^{)(-8y`rg}GO8PVjM?h*T_NpAJqM7+$# zE%+BPcf?kfs!ND%#3Ug{U~%?J>{L7&`j!L@d-&S+(9<9HGdXgsnQpEnUW_%S3q|vd z6)jS+_}WNkfYy|O_IIVXa91ipLm8g^w<7axZ7T-=E9Mkr4d_Fbb>^0$dgaAe`zb~P z)LQt*nN+y0qu~Dfz8}do*LhLbgM5hfMg0AB8%y-79JMJp&*#AmGY6ftz#kwScUDRO ziTQ`m?>|b9a`6llJYDVS49e-f+cFFM9RCG9K(`trK z01&%cE*KC}=WboJrC$Tlo4+QJB_W!d2iIf#tr4*ZDrZ{KOzrucw!4q+F~8u3;|Ax^ zu@!kM&vT?a1Becy|KhhZSwxcbM%S40XU?RY=|A3=Tn-&~S{w_Cu}ep>F7k-;UAd^@ zRhl@Ae51Z(zmN_bg!qC$U$%Tp{=DwU4m9N|=i}G3{GzTL$?2}Of>!snTsy3+rYgD6 zvkqc{X-Y+1*=c=#Z$5TB+qSq&e08|Iw75nEg6lyL2>rsq{?L7U@l7Yo`}h3Mbk@02 zPIl3_>|ew(?qQeY13(E(Lwg6%hQ+R-k(U4ZL(@y}Zv4EjRsRW~ZO8>5dR(xQN1_6U6$QnQ@57OEYs5v0 zC&9J@B;pDzv@hnqYoEMlwmdbMV{(GILI59HbJqrgVAJW%V|veWj2P zL+|<+`FVnNWPy0sCV@6($fe}hm>Sgz_GUZ2|9nZ8D6`y=^8TiQkP4E40q-Fo-bJv$zH)zR+_ z}zl?=lzMRkd{jrE^T6t2c^(w3wH6x3C#*aTxNM4vS^v*Fa^k_O#bc>CDaO zFr{`uXtdP?XEt4}uf7uey9fqq7Q~dx39T2tTQ0Mt*ZuNflX|uwQUOOIwR{U0KBio1Ktk`Gg8xV3^yq>)d z$Mf0vTQ+In_w#Z1kr^*_G)K|xk+~Ojv_R1s3?QGTRpbtK16 zSsZCw9Vw6>SrRq)io3A;YZ2uXROuRX2^MB-zH47T&{2g&s~aq}_HbTJR>yVJ{ArC3 zNmdLlM*VTkWhz1N@#rMn?ZnK^!#u;juH!DesjHz?tZ~MpI>7Wa%04t?H9J(_bK*4MmS27N9wOCDqq?=Nij9rXuBl4T{6rboQI8o>lqi{%Co}N3}x0 z_VZ8-UgXh)iK`F7OoA2yr$mojO7va3acMifS{s#Xhgc(5FE)wyFeD*?u_qW-F_-b( z3*!`;QPk0Z3?IQKGjp|)ejy+2E zXuy*h0u$%KY|BN@a)nV_3U_S(gp>HH-{rZVts>ke z$2>IpCo)`4N|roE@oTr!rpGdg+1W(c(iIz};D_H!L%~0gYz-s=c|v!`)=6gnVu7fj zQ86$^b@y+^(w3I2mYWMuaWowhgaj1!9@*k-!bk(z@S{!LAN6LK(wwN?$|sNv>+i3K zfjbn|Et;}B71ZutxB-UBaBh7@7A?DKo$Egx#9wO9Vy{sjc;pXWyL!t_G+#p_sWZ>H zuhy>R6=bUy7hKoMSHoQ&8#gbnn4p{W&+p<4GtbH;om0#yl%~T?$l{+)6P|W7L_hAX zYPT5ha@Gs-B5v?@Dk%c55QDv513#*C+D>I;wEc-5ih^>u^L;x&-u7c(E_rk=pX>T8 z2jG31f>lmj;;U zin)bbbaU1jWHDX^S_Hb^hSWd(W;*>0fhiW?Xy9e0?}d zCab;4(eURkYqlTzJlO7lZwD%W%Xh!uUVpp7EXYHrXSZm=hVYfW*WziFj^|GII1_fv zm^t7m>%aY4VaUGmU97lEw@qJCbZdau~Z&`AQ&5B<2OuMprxf0$;V)*5I=3|L- zKeJ`K6Ar8!d9nB>pTh0a#~jwT9Z5d=ObNZy68p}v!?EydfGdT2yxN0a>9Ljr98-1) zYrhldGdzdlFfy$KpGy>15CM%hlTg&9;y0tikTHV>$KAb@^7>+ZBcT z!QCvbW-$C27$Cc^T^2+0bC&Gm-gx+fw$0oVOMAqTX8ZHKTyle`+~N+pLl8$`#}ii9}j=p}{m=X)^`kabD*uU>Ud>{AByX%`kF0i#~jn6Zwzraq*= z(?Ta`EIY0s46?|qQG;XwOdR8R(h)~VMdnw>_VmCb!tCl&OH!caK^`&Z&=gVLS7KCR zjS>zNkm-rYeFmxrSgs)H?+QWkaa_Df^O0&XqHznoMrn2ozm>xZ03oywT38hxDKGT{ z#Ic}9-1y`1Oc2od{vPIoUNeXC*nqN@D3xC|6An%&Wr*^-e>lWE1EAyui!w|2>+?Hu z2uA(hsb=Pa4{fLeq>S`a{-8r1q=2Jy`X6W=x$hkli)`H~E~3ti4ajku13+PT8IYJV z+U$f=-X|aPRgd!enP!Te1#M31SQ7<Bb?hIjj_hb zBpv!>m<%2rl=!1~bnE%-Ng$arI~5?JVH`#D^FD{`A!#I%c>O!zorvQiz%Gat9>Htp zM~muE$1m2&m;eCeL-P6TkekTK)d$1CdW#6!Faz`(S7jAqgS$zzn!-t=<~$dAXA>Nz zNy0^^wn!rk1pvpHpygqBWfGu<_92sPZGhtpW=hDdASHbaRcF0E^bJSG@DW512SG%UI=aZ-628Z@QlNN0a08V!EyDgG0IS^E=&*c z2O`!f-l%P#lZe?wUOlXKL{#hE`(K6S1UG4r;D>Q!PU}SLE1_-auKYKJgSKmm1| zBF2(`EmCqobV()^1@7ipw-`;Mz<=^dk^hJb?p=3vUuDknJtGxar6^p?1M7}o<6)UH zMvzVJkwaRB_&6ej|MhAUqWckuiC>U0`<;VK6x(E(rjrDd;Km5$`-91#!mQ|v?}ktA zmqTWwk|+lFWV&Z9Te|6!EB+lCRmA7?*^QVCXKdiaRC1sS%MiJlMHDX54~nN!!1JYV zSoAMU1r0a>b4SX%i;I+k6OXwuh%N8la6~?#!Xw7)M5DNw1FOds1Zrm<(GRe9%CKt) zTMJT*<`)og1yO|~3hs@|;lUJQ(jeonpC@lSzLm?{omY~;dP|RDi8{i~XoIh3os)m} z3Byi^l+$B;777BM9NBTFWGR@MUzO&hPDP1`f;oLL4{AV^A8iYa5CWS(IhevICvopL z+VhBywIepnLByI`yT|NDepz-8XwR648hD0fL=Ck^1$^EC6bCgWdn>Zm-7@0p#uufQSC^)$z`Vb}PD%~+886$B$p!eI}0pXSw4%V5i z&+V#R|IE}r=8D+KC?JkKT2ER7#xE6l5Lp{=<~U;NgDQr($5ht4Ap??{BE2J>Fe-Z& zmCxab$P9)RP=dOJ1MV3##>)6IAVX^S@}m%Ql)lOi>=H%bIZ{vlzF?A|^m(V0Fs;W1 z+D|_Qk+87&m;|tMzlKq7$iUgqZFUn1?GO-th=rvVO;Q99;X-)<7R+->^J4{3jFC#j z!8dvTnA?=as!Xdz$n!aN{R%|ZpzPF|dHiQ-6e9+DzsCvi%?5;LEFjW4vbe=^y|pHx zfcT_A*&u-P4myZ~Px1byIO)dAb$n3*)RBt3j)D3l>k$hRkaxj$b0B1~PJs}s4KtPf z1@2R&|F`Op!W?8EnD=3lsy0)ATTPh69J-Q25_YH0%Ssjt!+<8eR(9H)uR6ECb<~gnW?=kvX@R!N}Phe&~?8Nd#>LrW?%LAGrAQ$6z+pPHRi+C>A zV9v5w@W@SOM_HAH20iouEizLB95NQth6(!p=p&ML2xLfb*di*W7YGIu*ucqekO^%u zXLcY9h<>0!DZ!jqpA<+83x&pX$((wnsjFhhNOk&I@PNgAF4)oV9K&C}1XJ^-M7V?$ zkb2#iTi70)P8m0L;d;!$*?dm*VI^%DjUuk`cS^%%VDl6K`g#%n=85BO8X@ui5C>;~ zPRol62|5Y~hVj(sO@@PT#sTqGaf-f!R>X3W5Us;bgBf^>1tA|H=zaXrE88QUQb*?k zFtWM)b&uLcnu-t~k?$sT1a&psA@bM|jObqfCv$cXo;h$SNSQ)}ADP`dHmV$0zP%Gj zgDZMMH!Dwi;$es?9+Dg87z0#cepcXXN7k;v8FS+kKnQ)CdB+i!(YH3pctLz5B>9^* zS{}?~LJ4sY)E?2s6o9x+QUe1#jD_8q$Rk9I`!q*4$360!kQ=s;R?=H(ky2s{@{?^w z8(@4h7_LVt7?pLyIqNs;3&8iHW(eGY|0anIUjuYZ3kON7M=-6(S3_<=#)2NV$!0qO zuF2+HLr00kq!alA+LZnxF`t)oA1m}Z=)So}$jO6K|21(h*kl8xp+#$|K&ns%OAWSw zr-#<-STyE>(PxYr%Q{ieClz|e6=_Q(!=G?Zfw(+yN3Q3U&JKbM*(9|q+&y$xM<2#5 zG{qO!Ey<6#H`A*D*bz7^8S<1L-!X=5{rST&iilrIj|AJNbwFjB>FGa%)hWwqbz*J2 zSLi+8%S#~7-8-9GOe=q01H6<5QKw2?2DVG)Q(!_;fSQdU>iU=eWKf@TCVcDX02=FO zzyP%h++@$nlD;+bPBPbI{WcI5GsOg<4o7MnpGBD6FgTcS#3JmU5Ecd9f6=Wux?n*& z3&LtAIXN$yH#~_l(__q#$_otIr#DayHlF?f1S2dV5D8zx?Y9{_a4g(U%WMIq!03ki=V z1?FJkKsHcT`Jn+H%jCEO(3ae7FruR&gL|z>MeG z`ttf)A#BBb7lSu(C>AK zKx83k*Bh^otagwm%IOpHb~v_ah_uJ^Ksr0+Uzqs=ozduiOi@XrtI)EdnsX??2(Q0E z7#J@xCN~a;$or8$#hjGlOEYFMVv8MG&;JU6E`HI|z|fM~#Zrg!Aai|$P`($!&z8)= zfRGWvV$Wrs6A47GuFS(Yje{VSlBF0uQ=-k#|48mekcIc|rw%-q2sj?gRD?*IB#9!p z8V+hIPb0WK@bhBj6vat^tfCnZ!6L??N}Qxg0M`I*Jr7;@S;)T=5TiPPWt@AiTFeOo zFM!+49qxHS<{-ZwEf9+mwPh!camKO9&|wxu>@Kbh7#mD zcSD5eaFLyiI66gHits5!1&cAHr!mG*?y@vQ8$h%ZIYclX0tVI8b-0>J!cfc-R9rWN z=R%x8jyOZ?%2PEKCn=mM^kX?{8|Se==uZnOB7sciv4_STsBJB9^8^nf7hRLu2Sv1* z$X$h_Hcw?dRx@B+U?7H?QOsSM7Zgdy<1trjX&6@2%29@j0rwptGMq}62nJ^po`3EJ z$X??368;MssmRcYz+EGP!>d*2-5gqf7Q;DVES4O-cP0YiBhbO{=W(=qwZuaN z3w`AY^E}THH!(Hmh=v)dxi|*<5y9ddgJl4>oyTxh-~;(S++*%wRG{wHh`n}MI!W`? z$502qJ47*hDI8?S^K#ng6#Izc;y8=p6$CN<^!+GLU7mO;58wn6!J-J(&jYm<=`q!& z0=2%$n49HcNCnFr1S!7fsm>ohnv8SbK{XB&5we76vjC9}1|}J$h{zc%7@|)^$b)Qi zFY$|Mlh?8+KguGNu zNUCc_v7ArWIEc*5981Vc}c$Q%I$YAU>Vw4)e-fa<Cf$t)x zq4Gq1H;C0_BvCV#IBFpb1>i@*O)^z=0|778M`+;?D=k_$A`{;U@t**(ii%6pAix+9 ziYy^!6GJ#hFu)N8_(BC0Nf292K6a z-N=Yd5_Gmv9H^PmnCQ50=D?Rt95jM7N5q3pBk_Zf=YF2}#EGRCV>fDAkf#FOSQNwN zX&bU5;>;OwzeX(7))wKoGFG!J5ds6WZ?!2oiX%4*iD0N4YWWLYj&k@sH*i$I?&LAX z&4AczAdL1A1t;)n(=bWkzX&&1;R?bsc+W(z6hmB+1!_mSaDpI!+ktv4om6m_I2RP> zQIMvw1CJB&NrE~oQKqqjXxm5ckr2VM$W=#25MO6%9w%~PJU_@)4CTbcZ01rVu|Ol@ z7+YMCpuWP?dV=hD+)bQF zA|^)$Q}jWBaMXDX*CFnTNMg09hDHuI5B&$a4%f{Co{9(=j)(z2Mx(^b@))J+qBoTA z`(Sv9azt}Ts4al09d=HZsN*8=I)uYGgTEc9+31|O9*{6p{Rlxm>Xq-LZlEHjcCUbs z{*t&=M1-y*bcyq4iJDPjEaHJ1xRHo`xLB?WhdcBzvIr^| zIvvl$c*k9r2YJpg#QW`h3%Xk3E_}HEVU#)n!`OszI8uA|m_&HSkYP>?s}m8c)fYHc z(i1Fq13-oe0tyKO@PwF6k^5+rj-RPX-yn?DftfTA=!G0k-GkhoA(9y!7Z@naMBvB( zoo<}MjtDhqA!35KH%t+j!`b6#s1967wRXZX-$ndG)GUX$hzMGSa1qdfM=4^m41TQ} z3Jfa&6^yHnH9}|yS4wRjKtB>@p3D$t6Bn3M2dy*A0tEd*CWAC#L6n4WMb*)`m^cWd zHG1&SV{{qt%hV(j^BCGeqBe}?32k&r5&gn(3Dwy!Cj554Oaiw+I8MqL(Q!$GJQ058 zrtMiLWR9hX@BB1Jj}{A}f0m-qfarxG2DwlQH%8^p9LG%p#AH;>LaEM&MHqw;l&Q|l z7jC2?bW~f!2Q)Cmg`V0pmZUJB+NbVB0q235I0+@OIi+gi7->c7c)sfjo@e+cI?Oyv zl8AT=B1ItLsY>B+FN{2lP(hHyVUnY=!MP=BmgexBoFG@*dBTV#uG%N+!e2<#!5Ly{ zB5+_~o_dal5hXy*0+C8JaVUKj5`B$Spz8auP#B>&(Os*3wGj*!xLM0Ad^ z#SQaNx)FwU5u!8CsbKA)fW(iX6Zi921`$(Hg_mJqp>>XkTDbZUfuQrQw)ZIKuP&Twm zi{flu&lKHa%)`t@x69SZZyvyLc4^@blWvUhTWy_`YK{=mzn=@Wj-nh$;))o+G9JXr zc5xo7xmAe}R-Hg|B0rb3F@WVPhle6FwRadz6CGp{28hMD6Tp$D^%|Kv_7=u!B|%`! zMYB}Rfh!cmN)Epk}7wL`q4lXBiv|=ElNDB&vK(=RGCN`k$5ES=Tn6MYryhgI)bZ>D;hUrOPG*T(LGjfd5TRE2iBVmK4#!8x zuTGsII822nxmuu?5@URr5H$-@j2uw{hgi+i25=DI*SZ*35(aNHAcArD+Of!Gg4X0H zhh43vt24-{o8?5!FdV@%2;dI8QZ28cMx+uQk|%xT@zKEGQ=oB1B0#jLE{%u~KB0Q3 zjkSUkuPVr5buQR}8>)62<*}3!VgyYDH?h10LlvN9!YHJp4L|5m|CEQpz0}+qB0Nt)H0|}kfq(Yt%!PH`f zuR zSZX247gEhd!rcmKrO932ci}%JvUM$7FNR~Q);zON?ek|uutd2nS?r`3PSPwuamGv? zYYElav5W|oGB*>E+7Acc0B#3rN|Ja81X7U^XOOv@`A*`iwP{bSc6$g8)mp#sGT(#u zM+5Ub7Gd~BFPp-1VKRgdk3iGsnSe_l5y3cgP*ZxSX(4hZT^^u+%U#!#g4hMA5e>co ztc1P;&X3I0Hrgmb>l8^uJSB-RSPkOsoE*jgUqpU@-a196txB3UqH{G}fB-+u(Fj~s zXJdF*TopK#` z)wZAj*2}!Wiy|&#=>;NC%TP2hwOJ|3kkv2?;UuelcA|BV6RB1PXkh9t4Fn??d$ZgR z{9G+o#GcA|0%w~FrY0fO;WD)~1wCxS)kQj4?jgz+F|j$xKuz2u1oK!9>!ogv`Wv|} z!aSE6yz)E`WVe$)L>%X;i3wHvl`n}-lGX(YA5KfEPOJGWjr>e<_@05ACL&{bn$fy# z5uk5|2b4M)ya7P-NF7gCCosKG5M6gQ8H#2NLy1@*4o0}?gb@M*wRe>f3qZuVgdZD5 z5j-oz5^?0G+(og*=w~zHT#!hdJXT#t!d(v@0!9RuU@VAz$3fZpM6gtC_mC<2Fm(t8 zQAX}^6{z_5S4Qh;g*ssw!6(3|q_)jxA5+5RcXZ1VUAqI^^x<^fY9dlM96Z zfb`)4(z+nESwLNd6eGdL@`XpdE^%c(Lq%GHs1i3*$( z!BnUO`^zj+v-j%iKG$PXh#a+;I0!=qZU8EOrW{a@!!34D8X!T1%YE9)D{5?%B7o(H z@l~(wh8gox)s_1$^N9;G5_QxSL7pbQRJVd9sI7=6A~gk{bK)i=iFhtVotLPwkVnhp z%KOY?bt_U#T=qccj}Fq${3PKL9(L$CQ3wyzP2v4{p5jLs4aeC?taA>J%?r z6(6CU+A%43s-kR1-IK?OEg)03e7Go6Pu*(e`BCTu7+)mf0?uSW1k2E~=0d8976$Kl82Gt>IM>cXy_A@B+N6E zaG;LM{!le=xP9Fhqypk`HQ23HJh?Z4hH1$<^I#k-ASugc%RfX;Q(|Ej4gd zA~nT_itFUL<0AqV~I7v(dL$8WD;i(&Sg1}32 zS8ea%eiSDO^@vjL#D0u=1^?R3r5~ogCl~@c7m>Ql!jK5YMO%1?y3`FDaMYNpBp6J6 z*CDzGxg$I%12;L#b9fo*E{nuRxiX9@5pmzPy7UH3QLRA2_ACryj0k9PW=VY_7(&e$U8_2!r{*_+2$KaIVHvt!+Pp@t?koc!PckPG z=#auZ#BdcO#s&}33qkP#MEk*~Hl2EMxa$%%;TPPy;62wb^~0SN+u_#Uw{vCbmtrA0kN*qN4~L=16sU4fkS|+)Zyln-q>^~k zWR9xU?!ba$9KlV<(bhK9&JAMROH|={sC+T%EV0|hU>>Tu zXc4RVOL&h-ggDqy+hqiGMJ(q6tw!z5P1Ow%2wKx9PB8|@EK3-1<2;<}41Hi6^C$=q zlKE-sIB~+_Sg0LKH2*Az3#8`BRJWtD8su2w z#KG)D`RmI7=Vuvi9tBYEbF_b|hY&O`6I{Ms(EfT!+djZkf7ibIse1^aEdOx(l;>aZ z?}F{RcVcc|d%3Fg`qqo)iJYB!FrpqK>b_p6+Q%hoi2U~DoqSYU5vbTbUi%clu{Lt`IX1cl?*`=cgO-;Zq*n zGxwvGX}?~z!(TVAU0fdP80vF*t^+b74|epPV!3-3r}r3(dKB!bD4$f?!MiN`N@yOg zIDC1!d^lIrs(p^Kd%|Mh{;}!aD${5DOxz}X>Jt9)EH5yBR6l&TKNOq#AbF7&yw+C| znCck{Zw8OHK=$cX_Q#+8`tIk;U*DDGq*DL&-9O~%h)VVk*?g*wAC$oB3*ch~c}>uP zl;zW_$ZLs~r~ToJE-USa*TUk;pO|`;y}lfEQ{Xc-;ineuzCv^i29HQ}32)w!8hfLRS4a7M_q@+-FJ+Lt z)3W#D43B!x$(%-|?z5=h3`qXS8o8~GUua|(DIMwM+cY`LYGX>I?GfRZ8TfO6W&|%p zA6|%A`2C(ju-^|U_WK!<`|)rtZJz*RC~BYDO`L(c1`S?2okh@By6eR-PhUH{iPs>>s(W<-~VQ{UbHW6QfI<=A-<5u4c$?=8GjW1)=_+-K?3brNkDiJNK`e4dg zm279htHegR)kh^wH_Ef^OqmrKwL{dm-De=1r;pK&R8tLuh@bC2BOvsBpxne01 zEv@=s#Mww{BT^!IPJA%stc%~I1+Nktb2SZwq&gzYkdWM``2Tt!XZbip*xF#^v#a5ANEzDJ*yCP$Lh@lg*myFtJDiJ-U zWJcX3-Rl{1c?|uXef1*Q=Yr~4Uc~g&=Z_}*W%i|!RQ66t%iVo4VipB;lu#miR(vqw zv~Vp){Yq1(@+iqJmJ-o&st;zIynue(g83pi!hiW}rbYeC$S&SlYJ|W2W<*|{N@`iR zGJ0b5#fY~4c%(+ct&E;leKDc6G4!~Zjg4HByUTi{yR6N+OA8lo)V(ux?drp>pcZK> zqi0oLjA-cG>~TO;mVc zM62bR%yE7x z`R2N@&W!USGT^*?nBi=8lGV+|CFJue(&xN>Fyr*-oZc9xXUOT(IsGwCe}*$FnomV7 zH;mva-pKS=Z@!q(7HThALuT&8M##I5X8hHW%v~Kvc&nos=VuGj&!eoLXK8m8|FGG! z`hyf3!cL~1T%Z2_>x>G=iV3c&W81fxkHLMxz7oC9rdl+Wg$7ME%HCAd%v=`c-K9$0 zV^TlOFznP0V$S%p`6z!Q)8~8rGDlk+2zQqx0&mu$0^M_f^7lMP( zB_DFV%sB=sjYs40bUdB`k5A+Abv(WS&t@x2c_g@Ww?Z1u%O?ZoYZyT+yIVn>1~;E6 zT6nfvc+*;VBrSYfE&ORMOzJ|ldm;4`t$>DA{XEA;=^lpt!fb~?*i)<7P&>bymY)Rs zzNk0sg%vxNQAVl0WS_hLeNGY5HW}>(8I!Fs#7!~U?Jy=AVW?YR^qXIdw!IKHylA(& zm~3+4r@E>3*ONX3!aKPE^WE2J6@<*$Y1eE7?O@BjSWkibk7oRJbChgP=I_Kt_`8pD z{M(}>OaPwK4;b5D=cqoO82pysC}zHh3d*6(32Z*-+i4`;d=1zFX~J#MTvXKp&Iw!%@tw0-3BhWI;>}-&8zr?O0S7%dQrMk zjqHp@vz_d%!1ZV{RhR3z<#(f>30l~A2V($JS{eHaw^B#RU#zrJf)m=P&onJIyW)ei zR;87m(Wgygu2d}v6QX^=302hRIW1laL9vBOK44To&hqRlN*-16AComC-V`UuI}at5ND+tWRE=k5RN$C*69J!Ag@n=H?W~yiahhWu{lk zS>3d=7@H6BSW5F&*B9&&_dgqG(OwU;DXZh2ro^k9PKVd81}f}j!**7dxRunYar@ms zk>kF$u*5}5r@-Z#33)}>pm&wfl2!+k`d4$dlR~w;jhYBL@n+MW8XBP98L^dAwo=b#X2$xtpxd$)(zCXo=9vp>2^RVyEOJve zk4)!{@Ofs0C7G}sAuP=Z=|x#glx0CE%pI-@s6=^1jCbk1^&)1Xwz zb{#RRnN{gV9kZ#SN$D0HGpLC@i!wLg)Y7Kz6RF>3NQlQc^#?eOj&A0=im*$nO+BEg z=5|K4NjAj!mRlhm<;G<0B|j1Fo=}PVOsBaUC%V~N%6_k_M33wGyAk0>I^)MN%8#?0 z5`Gopx)-_G%RS|0vw>t^gZ}qJg>z;vJW;&+{kHkxcNGx!jYLz zDXPSMGW;~hAQ>e}KBph>w7(ir5jS?H$ePcjy{P65Ecj&_{=<>bU8ux80`=20 z!%pVSh8FC;iuBlCKg@B~HG!?J2RwB>&p;eUUqv)*CL`$iZf^cuLE?k?;MJ4sv)_Lu zsj#7}a%p6zRoTq0e_aHQ6{WmyghmulpNyDgv8D1tM9+#3rku|fq|c+Q&!)65Y*$+_ zE3r{>^^wB=Y^lKWqy}atu&C{$mE5#$?@m<8-AT{5I~fvgazu)R*^0j482$OPkrt#y zof{cF_xys+IpM*gh!3M>d@wEKk7Q$i9I^68GxOG@UGcXQi#AyX9x2GLD zS}AcOKcUY}Dl^qy1X-IGF`~eWFye%YHiPI{OwWD(ILluj7$;SNR$$24{yf805vEa8 z;yz0?H;qEMt!;Qu+w2W(w7c1Kx3U@B$40w}O?L;I!R>3Ld)G8KuFc=Ih7?Q)%Fdyy z#E`4|eU^}Zt=q(fZsu3HA!WhYj&$1mMrO!+`C>*}5_a|6$c)gIb31x22`Rq8Rb+&8 z)*m2WlQ(ujp5fJad)Dkv=22ajyhqjjILGsuK=OIO@Oc)#*b}y?p3@I$)UPvClw18Y z_xa7=0!t?_VCsIJV`~tjcPD^0q;<%FGSZnhtI-<;Lf*D;1M+k@K2WJltA_1)>cj<>T1}&L- zb>+yVm1fsf?$ujthn@LE)e;ww+FZv@jV=vSgK2LL~J`s%8rtTA|-)2ZChqg5*w&#y) zA5P7N?W%ISA@#fWVtv-TkJJ3*W4$AXdd-gXUQ&q-(aY`&j<8=oPit{xqF|-DvCrvy zoabNXs7g`ozt7~ctWLXKLEyX>9FX6CHqwH3j~f=0JA{X#dQEN;80oS-?K4SMqm$BW z(0wx^{~*6mnIWR5#RntK9XsqO({1-+dhYW_BmUM&dSS-`1=SvKpjQG7gkCRALL(jR z)SA+SKz@+N{i$K3F>ENRSK}sukuK^#McGjGIE(3ZIDa(auO;CM&Kns$v-)C2OS+c% zR%nEH{$l2H%I0`#Z#NT`O`|Wb*QcM#NVQX0kT%QRi|G|Pf1Kf`+;XqE-+umPdy6~m zM{lz?y~qA2_e{ONd?ZkwB?gS=-{%MqlJJsRmwZ5_ew^c}Ew{@KBnRv@Wy>t}jR5`Y z07mZu9E1~ns0 zf2hPg3iZ=G!)CiZ$y%E&W-+Z3e;x`iYZyLVe3`p90 z8_*tWf>Y2J)KTAjp3&lZYG7CF2_8aEuCK%W*Lf8z?;9J6Y80W|J9e9%INvuEozTT( z{!gXv1_g_*H(zH|D7O2GutIfC->2+;ouw)%Jmrw2 zoMWI64DsjTpDFmyLxPzNd-aagLU)=*xtni?D#_+_5R{vUZ-sP}=T9@tRo+n6kI(7* zeBG~eR0JPE^8on#>)*7fPZ#bP#CarFN8{-jI6Ab4cC=6Im^`jSeO5>RppMZ~I(5VN z2Xi8ox=*8inkHojONsi6SYo*hkar@x}|^qzh-4ftw->9pyWR^maMA=vNK;8DOE)zmVsC^<7Dz zDm{`i8or{$uY7_k$}mtE2A?6afn;c?2{k5!!!_BL3E%Q+kY_Vxq;2{(%%QdmWBeU~ zwDXuEGsmPe-+>#}z5@|z_(3Ty|5QuJh;Zf7P&)5NWS7zEEsf-XTt1MeU(-jo+bMrf;j)U3fT!;m8z z1{(pAE;SLhTlT%|Pv~^YA#6y*?17Qf6^8-WN%BCT09AxxCcUH~9LT@?!d64rY+vQ^ zZQ7=<)2!{Rk+3`=4~VADIF-m7@u&6|Nmc=Oc`}Xk{_y$ZjdnideV&Qr43OTrPys*k z|1V#}{}Wu!;xpMos0#w)_+Dc`Rd6YG6{(A!BhRVVp@>Feb`{BR|IpU;(3E>?8H!4- z3xrYyD$1>L`ZOkjnZ$_Hpa4{643)r7`Fx@;oE&e;fcNRsi}qC87y;*Tf1~ujFPnmQ zziz5y@ukZ(bo^vc=M8@c)$oju=c58XsE_6klX;HWg9{YYHh~e_&+85v=+SD%=jO2( z#JOaTMgP6+UfGNRUqLtbg7{nr+tP#QwxA&j%T*BL;5r!B;=s3bff1AZRjc@_CA@Xc| zo(+a)OXS)5JX;LU%i@`UIH1S;;vWcziL^AxYS#O4u$)i2e{CyY}1;J)U|5 zSd^m;M@6aX2?Eh=kk_`Q6K+AmXl0U`8@IKIH#WERm*Q|JnP;Mt_VO(jh+p7yv4E@g^hY`)2i;0&&4 ziDEoXds#~W|B(Nnj(R4D#W-1Ugc~~H1|-}d3Ac2@El3#c6g-d`vj?{wdx$q-5A9*7 ziud^$-ZhRu*^EX3fmv+*uI2wMO@t93sK@ZZN!y>w>d+MpS&{}qQgj(kLf|;EI0&LP z&n8=r@U^T&4r(!t2G1)~;E)vC11YKttkS~3KoOm}U~4%_RpeijZ&4Er2{Fr&a@zcm zeHWP~`vf@#7?fj_aW0oW3h}H0ce9uTEC`1<`RrK%zV+0Ch?~V1v%Q5;cFX=B7iGrAA0fAW_ z8oJ`%gn|ThIerA0qj7i_5F_`Kin)F%!kI>a=hY*mM7Zy^vN5WJ|V`Xf9up633GNn)}r@|v0A8uWd{hW zKnjln1ckv18UGeWAnia|rZpPSblR3s&lCG+Bud0k+6?HBP zn#jVZ4MqBRO95&S&Dcn(yHZD1gqp!5US%;Bv!=?1QigfVBxeq z49ur@kKJ2}l2rMjn#WgrTAc?ge7Ux#Bl?(*wlBbQV^BDYT^P`;iKH$kh4Prld2~sS zRMsp%pS$))D)|WaG>mqd6tUuVtsWzWFUlyyD@*)k+#iPUyPU)spx9p7BHWT-*lALF zR0n4P?91rGogO$2a*fFX*b}j|+Qg4ji4_i6m-wL0XiLUy#_MR08N&_-7vZ*&*-n#! z9fU2yeJP8>GAI^f1$+OcY(%j<`7Yyc*RDPPOp_K2=SU6uQKj$2$(;rN4eGMWJKinG zD)>0SbpavyuqXKf{-a8d*7mAuKIQ{sW!_{>RjsXIFBYWMbD}CfmCdnOD|HS6X`D;R z;)X9pw_4m31Qa)wQpLSi*`a(}tImi5$uo+n`l|n})iy?ebd5!1O}z2t?ZIiaGBXaS z&vc-w-BcDm#<5XUcmdv_I zrQLYE8xrw#D)F^Pe2pe%iS^9fdS;Y*UNP{220hdP4=WQdx^yqPq+E3R5DOY!QHNFx z`|T?T_349O!QdFU?_puACt%^r7#t;iHip4p%UJAd84wBeB@4#c+~IFpK00;s$D8+Y zrhFy<9DFJs_tm?b_PL9gQij25c3@(o;Q3Hv4XaPc^%q%|mi1jKxbx2H|pyZ!*QvL~Q z`6odP_{JsSgL3A#FEZXeyaJPe2DRK9t1pqUX*wUnrAZAega&^!4;0&m`=I1$OhC0S=^%RyS8h*E1q0+}JW_&hOEM6A6u9m(I4FnIzB|HV%)d($Dd<>0_S2t z+rF*qw#krV6W@JIa3{^ngO(Q8L22#&$5LNIS9whI?5ASxrvG(Nb$A^RWjxhR!k*>d z8q(_}n-KY2sG!sOPBz0uZbOmg5QOVO4KIqz&*J|JLur+Zb5236?|##2JMmeBor`_> zwcIlsH?aIz@Ub-?5{s{mnVaM9od93W;AgSie5e1tVELOsiM`8Wr>tI!?u7TfRBUCx z;`S?z-bs_pO?W?=WS&J%W&hIdm4|<8UJ=Hx&G>`5KiMIaZM-p5`4iWDApE0%AZ?^q7|R-|0DjU)({ARB6nDownWMb)1XTtS!i6K=f;l2iARl0e%iu06vO|>*%nW)I4mZ|$KMzrvDP?XBSVvgRfZUwnbm{&#br0GgyoC=%X fqJC3qy+>tJI=xA!6!H#xsD0q*tBd~wGsPHoK2<#- diff --git a/docs/build/html/python/_autosummary/mlx.core.Device.html b/docs/build/html/python/_autosummary/mlx.core.Device.html index b4e33ed4f..28138d1cb 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.29.2 documentation + mlx.core.Device — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -939,7 +939,7 @@ document.write(`

    A device to run operations on.

    -__init__(self, type: mlx.core.DeviceType, index: int = 0) → None#
    +__init__(self, type: mlx.core.DeviceType, index: int = 0) → None#

    Methods

    diff --git a/docs/build/html/python/_autosummary/mlx.core.Dtype.html b/docs/build/html/python/_autosummary/mlx.core.Dtype.html index 2254bedf2..e49350b4c 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.29.2 documentation + mlx.core.Dtype — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html b/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html index 114e0c5f0..82fcd9ed4 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.29.2 documentation + mlx.core.DtypeCategory — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/_autosummary/mlx.core.abs.html b/docs/build/html/python/_autosummary/mlx.core.abs.html index b3ebe12f6..5901b1235 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.29.2 documentation + mlx.core.abs — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

    mlx.core.abs#

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

    Element-wise absolute value.

    Parameters:
    diff --git a/docs/build/html/python/_autosummary/mlx.core.add.html b/docs/build/html/python/_autosummary/mlx.core.add.html index 04da49534..976fedb50 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.29.2 documentation + mlx.core.add — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

    mlx.core.add#

    -add(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
    +add(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

    Element-wise addition.

    Add two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

    diff --git a/docs/build/html/python/_autosummary/mlx.core.addmm.html b/docs/build/html/python/_autosummary/mlx.core.addmm.html index 812551b2a..b4c54a122 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.29.2 documentation + mlx.core.addmm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

    mlx.core.addmm#

    -addmm(c: array, a: array, b: array, /, alpha: float = 1.0, beta: float = 1.0, *, stream: None | Stream | Device = None) → array#
    +addmm(c: array, a: array, b: array, /, alpha: float = 1.0, beta: float = 1.0, *, stream: None | Stream | Device = None) → array#

    Matrix multiplication with addition and optional scaling.

    Perform the (possibly batched) matrix multiplication of two arrays and add to the result with optional scaling factors.

    @@ -942,9 +942,9 @@ with optional scaling factors.

  • c (array) – Input array or scalar.

  • a (array) – Input array or scalar.

  • b (array) – Input array or scalar.

    • -

  • alpha (float, optional) – Scaling factor for the +

  • alpha (float, optional) – Scaling factor for the matrix product of a and b (default: 1)

    • -

  • beta (float, optional) – Scaling factor for c (default: 1)

    • +

  • beta (float, optional) – Scaling factor for c (default: 1)

    • Returns:
    diff --git a/docs/build/html/python/_autosummary/mlx.core.all.html b/docs/build/html/python/_autosummary/mlx.core.all.html index 38bbffde7..b6738e211 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.29.2 documentation + mlx.core.all — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

    mlx.core.all#

    -all(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
    +all(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

    An and reduction over the given axes.

    Parameters:

    • a (array) – Input array.

      • -

    • axis (int or list(int), optional) – Optional axis or +

    • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

      • -

    • keepdims (bool, optional) – Keep reduced axes as +

    • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

    diff --git a/docs/build/html/python/_autosummary/mlx.core.allclose.html b/docs/build/html/python/_autosummary/mlx.core.allclose.html index 04ac068a3..a379d50c8 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.29.2 documentation + mlx.core.allclose — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

    mlx.core.allclose#

    -allclose(a: array, b: array, /, rtol: float = 1e-05, atol: float = 1e-08, *, equal_nan: bool = False, stream: None | Stream | Device = None) → array#
    +allclose(a: array, b: array, /, rtol: float = 1e-05, atol: float = 1e-08, *, equal_nan: bool = False, stream: None | Stream | Device = None) → array#

    Approximate comparison of two arrays.

    Infinite values are considered equal if they have the same sign, NaN values are not equal unless equal_nan is True.

    The arrays are considered equal if:

    @@ -946,9 +946,9 @@ broadcasting.

    • a (array) – Input array.

    • b (array) – Input array.

      • -

    • rtol (float) – Relative tolerance.

      • -

    • atol (float) – Absolute tolerance.

      • -

    • equal_nan (bool) – If True, NaNs are considered equal. +

    • rtol (float) – Relative tolerance.

      • +

    • atol (float) – Absolute tolerance.

      • +

    • equal_nan (bool) – If True, NaNs are considered equal. Defaults to False.

    diff --git a/docs/build/html/python/_autosummary/mlx.core.any.html b/docs/build/html/python/_autosummary/mlx.core.any.html index 6df618871..74f4413a6 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.29.2 documentation + mlx.core.any — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

    mlx.core.any#

    -any(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
    +any(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

    An or reduction over the given axes.

    Parameters:

    • a (array) – Input array.

      • -

    • axis (int or list(int), optional) – Optional axis or +

    • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

      • -

    • keepdims (bool, optional) – Keep reduced axes as +

    • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

    diff --git a/docs/build/html/python/_autosummary/mlx.core.arange.html b/docs/build/html/python/_autosummary/mlx.core.arange.html index f636ae4bf..bb8b339e3 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.29.2 documentation + mlx.core.arange — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,9 +932,9 @@ document.write(`

    mlx.core.arange#

    -arange(start: int | float, stop: int | float, step: None | int | float, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array#
    +arange(start: int | float, stop: int | float, step: None | int | float, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array#
    -arange(stop: int | float, step: None | int | float = None, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array
    +arange(stop: int | float, step: None | int | float = None, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array

    Overloaded function.

    1. arange(start : Union[int, float], stop : Union[int, float], step : Union[None, int, float], dtype: Optional[Dtype] = None, *, stream: Union[None, Stream, Device] = None) -> array

      diff --git a/docs/build/html/python/_autosummary/mlx.core.arccos.html b/docs/build/html/python/_autosummary/mlx.core.arccos.html index 8049eadd1..746d68478 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.29.2 documentation + mlx.core.arccos — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.arccos#

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

      Element-wise inverse cosine.

      Parameters:
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arccosh.html b/docs/build/html/python/_autosummary/mlx.core.arccosh.html index 3ef3d0c93..d39f3cae2 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.29.2 documentation + mlx.core.arccosh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.arccosh#

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

      Element-wise inverse hyperbolic cosine.

      Parameters:
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsin.html b/docs/build/html/python/_autosummary/mlx.core.arcsin.html index 835635a57..4be3ce315 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.29.2 documentation + mlx.core.arcsin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.arcsin#

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

      Element-wise inverse sine.

      Parameters:
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsinh.html b/docs/build/html/python/_autosummary/mlx.core.arcsinh.html index dd8f2f3ff..2976eb7dd 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.29.2 documentation + mlx.core.arcsinh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.arcsinh#

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

      Element-wise inverse hyperbolic sine.

      Parameters:
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan.html b/docs/build/html/python/_autosummary/mlx.core.arctan.html index ade61a5a5..32a01bccd 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.29.2 documentation + mlx.core.arctan — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.arctan#

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

      Element-wise inverse tangent.

      Parameters:
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan2.html b/docs/build/html/python/_autosummary/mlx.core.arctan2.html index db2b9fed9..de9fba2fa 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.29.2 documentation + mlx.core.arctan2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.arctan2#

      -arctan2(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#
      +arctan2(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#

      Element-wise inverse tangent of the ratio of two arrays.

      Parameters:
      diff --git a/docs/build/html/python/_autosummary/mlx.core.arctanh.html b/docs/build/html/python/_autosummary/mlx.core.arctanh.html index f9e392dd7..c1345d440 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.29.2 documentation + mlx.core.arctanh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.arctanh#

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

      Element-wise inverse hyperbolic tangent.

      Parameters:
      diff --git a/docs/build/html/python/_autosummary/mlx.core.argmax.html b/docs/build/html/python/_autosummary/mlx.core.argmax.html index 76ba54bf8..cd871f1e8 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.29.2 documentation + mlx.core.argmax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

      mlx.core.argmax#

      -argmax(a: array, /, axis: None | int = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
      +argmax(a: array, /, axis: None | int = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

      Indices of the maximum values along the axis.

      Parameters:

      • a (array) – Input array.

        • -

      • axis (int, optional) – Optional axis to reduce over. If unspecified +

      • axis (int, optional) – Optional axis to reduce over. If unspecified this defaults to reducing over the entire array.

        • -

      • keepdims (bool, optional) – Keep reduced axes as +

      • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

      diff --git a/docs/build/html/python/_autosummary/mlx.core.argmin.html b/docs/build/html/python/_autosummary/mlx.core.argmin.html index c59bcd55f..167b99391 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.29.2 documentation + mlx.core.argmin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

      mlx.core.argmin#

      -argmin(a: array, /, axis: None | int = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
      +argmin(a: array, /, axis: None | int = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

      Indices of the minimum values along the axis.

      Parameters:

      • a (array) – Input array.

        • -

      • axis (int, optional) – Optional axis to reduce over. If unspecified +

      • axis (int, optional) – Optional axis to reduce over. If unspecified this defaults to reducing over the entire array.

        • -

      • keepdims (bool, optional) – Keep reduced axes as +

      • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

      diff --git a/docs/build/html/python/_autosummary/mlx.core.argpartition.html b/docs/build/html/python/_autosummary/mlx.core.argpartition.html index 160c76cbd..742a1b886 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.29.2 documentation + mlx.core.argpartition — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.argpartition#

      -argpartition(a: array, /, kth: int, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#
      +argpartition(a: array, /, kth: int, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#

      Returns the indices that partition the array.

      The ordering of the elements within a partition in given by the indices is undefined.

      @@ -940,12 +940,12 @@ is undefined.

      Parameters:

      • a (array) – Input array.

        • -

      • kth (int) – Element index at the kth position in the output will +

      • kth (int) – Element index at the kth position in the output will give the sorted position. All indices before the kth position will be of elements less or equal to the element at the kth index and all indices after will be of elements greater or equal to the element at the kth index.

        • -

      • axis (int or None, optional) – Optional axis to partition over. +

      • axis (int or None, optional) – Optional axis to partition over. If None, this partitions over the flattened array. If unspecified, it defaults to -1.

      diff --git a/docs/build/html/python/_autosummary/mlx.core.argsort.html b/docs/build/html/python/_autosummary/mlx.core.argsort.html index 52c06f64f..6f36958c8 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.29.2 documentation + mlx.core.argsort — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

      mlx.core.argsort#

      -argsort(a: array, /, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#
      +argsort(a: array, /, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#

      Returns the indices that sort the array.

      Parameters:

      • a (array) – Input array.

        • -

      • axis (int or None, optional) – Optional axis to sort over. +

      • axis (int or None, optional) – Optional axis to sort over. If None, this sorts over the flattened array. If unspecified, it defaults to -1 (sorting over the last axis).

      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 0047c9bf0..dec3d7d1d 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.29.2 documentation + mlx.core.array.T — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 e10dc953a..2d64735fa 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.29.2 documentation + mlx.core.array.abs — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.abs#

      -array.abs(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.abs(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See abs().

      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 e01f2d932..ecc09d46e 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.29.2 documentation + mlx.core.array.all — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.all#

      -array.all(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.all(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See all().

      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 edc1246cd..b9dd6b4b9 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.29.2 documentation + mlx.core.array.any — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.any#

      -array.any(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.any(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See any().

      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 eb68b7dda..28323b373 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.29.2 documentation + mlx.core.array.argmax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.argmax#

      -array.argmax(self, axis: Optional[int] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.argmax(self, axis: Optional[int] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See argmax().

      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 977fbff56..11caad17c 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.29.2 documentation + mlx.core.array.argmin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.argmin#

      -array.argmin(self, axis: Optional[int] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.argmin(self, axis: Optional[int] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See argmin().

      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 9a78073e7..db8bada76 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.29.2 documentation + mlx.core.array.astype — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.astype#

      -array.astype(self, dtype: Dtype, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.astype(self, dtype: Dtype, stream: Optional[Union[Stream, Device]] = None) → array#

      Cast the array to a specified type.

      Parameters:
      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 a3180fe0c..29dc2d5a6 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.29.2 documentation + mlx.core.array.at — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 09f5dcb41..84a547617 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.29.2 documentation + mlx.core.array.conj — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.conj#

      -array.conj(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.conj(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See conj().

      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 0486a7b41..627a14e08 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.29.2 documentation + mlx.core.array.cos — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.cos#

      -array.cos(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.cos(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See cos().

      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 b35ef12dc..bd0bb46e0 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.29.2 documentation + mlx.core.array.cummax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.cummax#

      -array.cummax(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.cummax(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#

      See cummax().

      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 783e44d97..c8b366fae 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.29.2 documentation + mlx.core.array.cummin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.cummin#

      -array.cummin(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.cummin(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#

      See cummin().

      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 b40a12e9b..09bd817ab 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.29.2 documentation + mlx.core.array.cumprod — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.cumprod#

      -array.cumprod(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.cumprod(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#

      See cumprod().

      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 a5140718a..fea2f8277 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.29.2 documentation + mlx.core.array.cumsum — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.cumsum#

      -array.cumsum(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.cumsum(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#

      See cumsum().

      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 ed3fc209d..33ea52efb 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.29.2 documentation + mlx.core.array.diag — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.diag#

      -array.diag(self, k: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.diag(self, k: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array#

      Extract a diagonal or construct a diagonal matrix.

      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 a43f667d2..48b8e7c55 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.29.2 documentation + mlx.core.array.diagonal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.diagonal#

      -array.diagonal(self, offset: int = 0, axis1: int = 0, axis2: int = 1, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.diagonal(self, offset: int = 0, axis1: int = 0, axis2: int = 1, stream: Optional[Union[Stream, Device]] = None) → array#

      See diagonal().

      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 f7d79fb2f..87583d5cb 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.29.2 documentation + mlx.core.array.dtype — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 a7590e6e9..ab4981212 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.29.2 documentation + mlx.core.array.exp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.exp#

      -array.exp(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.exp(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See exp().

      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 3177e5ab5..6f8a40d49 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.29.2 documentation + mlx.core.array.flatten — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.flatten#

      -array.flatten(self, start_axis: int = 0, end_axis: int = -1, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.flatten(self, start_axis: int = 0, end_axis: int = -1, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See flatten().

      diff --git a/docs/build/html/python/_autosummary/mlx.core.array.html b/docs/build/html/python/_autosummary/mlx.core.array.html index 9ba40833c..750f5497c 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.29.2 documentation + mlx.core.array — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -939,7 +939,7 @@ document.write(`

      An N-dimensional array object.

      -__init__(self: array, val: scalar | list | tuple | ndarray | array, dtype: Dtype | None = None)#
      +__init__(self: array, val: scalar | list | tuple | ndarray | array, dtype: Dtype | None = None)#

      Methods

      @@ -1036,7 +1036,7 @@ document.write(`

      See reciprocal().

      reshape(self, *shape[, stream])

      -

      Equivalent to reshape() but the shape can be passed either as a tuple or as separate arguments.

      +

      Equivalent to reshape() but the shape can be passed either as a tuple or as separate arguments.

      round(self[, decimals, stream])

      See round().

      @@ -1069,7 +1069,7 @@ document.write(`

      See swapaxes().

      tolist(self)

      -

      Convert the array to a Python list.

      +

      Convert the array to a Python list.

      transpose(self, *axes[, stream])

      Equivalent to transpose() but the axes can be passed either as a tuple or as separate arguments.

      diff --git a/docs/build/html/python/_autosummary/mlx.core.array.imag.html b/docs/build/html/python/_autosummary/mlx.core.array.imag.html index a8631a7f2..fe9ab04d7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.imag.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.imag.html @@ -8,7 +8,7 @@ - mlx.core.array.imag — MLX 0.29.2 documentation + mlx.core.array.imag — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 1b4704a9a..82c8900fa 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.29.2 documentation + mlx.core.array.item — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 040af0b2d..37f63bba4 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.29.2 documentation + mlx.core.array.itemsize — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 f6c55d0a3..f0d63a6d9 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.29.2 documentation + mlx.core.array.log — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.log#

      -array.log(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.log(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See log().

      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 323a53d7b..68a75ca4f 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.29.2 documentation + mlx.core.array.log10 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.log10#

      -array.log10(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.log10(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See log10().

      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 e8148cc96..99727a0a8 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.29.2 documentation + mlx.core.array.log1p — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.log1p#

      -array.log1p(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.log1p(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See log1p().

      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 56505a10c..0355c0d7d 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.29.2 documentation + mlx.core.array.log2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.log2#

      -array.log2(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.log2(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See log2().

      diff --git a/docs/build/html/python/_autosummary/mlx.core.array.logcumsumexp.html b/docs/build/html/python/_autosummary/mlx.core.array.logcumsumexp.html index 43b8d2826..f43f87edc 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.logcumsumexp.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.logcumsumexp.html @@ -8,7 +8,7 @@ - mlx.core.array.logcumsumexp — MLX 0.29.2 documentation + mlx.core.array.logcumsumexp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.logcumsumexp#

      -array.logcumsumexp(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.logcumsumexp(self, axis: Optional[int] = None, *, reverse: bool = False, inclusive: bool = True, stream: Optional[Union[Stream, Device]] = None) → array#

      See logcumsumexp().

      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 3d17b67bf..3506feb79 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.29.2 documentation + mlx.core.array.logsumexp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.logsumexp#

      -array.logsumexp(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.logsumexp(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See logsumexp().

      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 51b69fe70..72ae30796 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.29.2 documentation + mlx.core.array.max — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.max#

      -array.max(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.max(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See max().

      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 c846cb602..5824f7b01 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.29.2 documentation + mlx.core.array.mean — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.mean#

      -array.mean(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.mean(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See mean().

      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 55c612900..7b38e75ee 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.29.2 documentation + mlx.core.array.min — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.min#

      -array.min(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.min(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See min().

      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 d372ba913..2803298fb 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.29.2 documentation + mlx.core.array.moveaxis — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.moveaxis#

      -array.moveaxis(self, source: int, destination: int, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.moveaxis(self, source: int, destination: int, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See moveaxis().

      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 977b71abb..f460d4056 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.29.2 documentation + mlx.core.array.nbytes — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 29be2140d..f0e219be0 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.29.2 documentation + mlx.core.array.ndim — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 a9fede5ee..b1c01ccb7 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.29.2 documentation + mlx.core.array.prod — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.prod#

      -array.prod(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.prod(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See prod().

      diff --git a/docs/build/html/python/_autosummary/mlx.core.array.real.html b/docs/build/html/python/_autosummary/mlx.core.array.real.html index e87719303..78dd8ecbb 100644 --- a/docs/build/html/python/_autosummary/mlx.core.array.real.html +++ b/docs/build/html/python/_autosummary/mlx.core.array.real.html @@ -8,7 +8,7 @@ - mlx.core.array.real — MLX 0.29.2 documentation + mlx.core.array.real — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 a8b36f3c0..8b01d005e 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.29.2 documentation + mlx.core.array.reciprocal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.reciprocal#

      -array.reciprocal(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.reciprocal(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See reciprocal().

      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 f24a57793..d7d2a387d 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.29.2 documentation + mlx.core.array.reshape — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,9 +932,9 @@ document.write(`

      mlx.core.array.reshape#

      -array.reshape(self, *shape, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.reshape(self, *shape, stream: Optional[Union[Stream, Device]] = None) → array#

      Equivalent to reshape() but the shape can be passed either as a -tuple or as separate arguments.

      +tuple or as separate arguments.

      See reshape() for full documentation.

      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 6162cdad2..5956dd3f1 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.29.2 documentation + mlx.core.array.round — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.round#

      -array.round(self, decimals: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.round(self, decimals: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See round().

      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 8b12363b5..c88a5017c 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.29.2 documentation + mlx.core.array.rsqrt — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.rsqrt#

      -array.rsqrt(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.rsqrt(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See rsqrt().

      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 f4b868b6d..ffa7059a9 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.29.2 documentation + mlx.core.array.shape — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -939,7 +939,7 @@ document.write(`

      A tuple containing the sizes of each dimension.

      Return type:
      -

      tuple(int)

      +

      tuple(int)

      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 56b8bff7c..f7119258b 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.29.2 documentation + mlx.core.array.sin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.sin#

      -array.sin(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.sin(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See sin().

      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 97dd8c27a..5b370e250 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.29.2 documentation + mlx.core.array.size — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 53503db03..cfd94592d 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.29.2 documentation + mlx.core.array.split — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.split#

      -array.split(self, indices_or_sections: Union[int, tuple[int, ...]], axis: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → list[array]#
      +array.split(self, indices_or_sections: Union[int, tuple[int, ...]], axis: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → list[array]#

      See split().

      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 d0d9d89f1..69bb9b016 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.29.2 documentation + mlx.core.array.sqrt — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.sqrt#

      -array.sqrt(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.sqrt(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See sqrt().

      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 ddfd978c4..8965ae1ba 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.29.2 documentation + mlx.core.array.square — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.square#

      -array.square(self, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.square(self, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See square().

      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 5b05720db..60c2fcfe9 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.29.2 documentation + mlx.core.array.squeeze — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.squeeze#

      -array.squeeze(self, axis: Optional[Union[int, Sequence[int]]] = None, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.squeeze(self, axis: Optional[Union[int, Sequence[int]]] = None, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See squeeze().

      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 422995d2f..f77e673bb 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.29.2 documentation + mlx.core.array.std — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.std#

      -array.std(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, ddof: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.std(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, ddof: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See std().

      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 90a93bac5..77a5fcb8f 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.29.2 documentation + mlx.core.array.sum — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.sum#

      -array.sum(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.sum(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See sum().

      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 660ce57cb..dbd7412cd 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.29.2 documentation + mlx.core.array.swapaxes — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.swapaxes#

      -array.swapaxes(self, axis1: int, axis2: int, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.swapaxes(self, axis1: int, axis2: int, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See swapaxes().

      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 0753ede42..1705b3c3b 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.29.2 documentation + mlx.core.array.tolist — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`
      array.tolist(self) → list_or_scalar#
      -

      Convert the array to a Python list.

      +

      Convert the array to a Python list.

      Returns:

      The Python list.

      @@ -945,7 +945,7 @@ list of lists.

      Return type:
      -

      list

      +

      list

      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 028a7fc78..ceb44bb49 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.29.2 documentation + mlx.core.array.transpose — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.transpose#

      -array.transpose(self, *axes, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.transpose(self, *axes, stream: Optional[Union[Stream, Device]] = None) → array#

      Equivalent to transpose() but the axes can be passed either as a tuple or as separate arguments.

      See transpose() for full documentation.

      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 bc4198fa8..75840509b 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.29.2 documentation + mlx.core.array.var — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.var#

      -array.var(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, ddof: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.var(self, axis: Optional[Union[int, Sequence[int]]] = None, keepdims: bool = False, ddof: int = 0, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See var().

      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 ace0d6420..5a7f386fc 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.29.2 documentation + mlx.core.array.view — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array.view#

      -array.view(self, dtype: Dtype, *, stream: Optional[Union[Stream, Device]] = None) → array#
      +array.view(self, dtype: Dtype, *, stream: Optional[Union[Stream, Device]] = None) → array#

      See view().

      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 603d80504..7b3e0e214 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.29.2 documentation + mlx.core.array_equal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.array_equal#

      -array_equal(a: scalar | array, b: scalar | array, equal_nan: bool = False, stream: None | Stream | Device = None) → array#
      +array_equal(a: scalar | array, b: scalar | array, equal_nan: bool = False, stream: None | Stream | Device = None) → array#

      Array equality check.

      Compare two arrays for equality. Returns True if and only if the arrays have the same shape and their values are equal. The arrays need not have @@ -942,7 +942,7 @@ the same type to be considered equal.

      • a (array) – Input array or scalar.

      • b (array) – Input array or scalar.

        • -

      • equal_nan (bool) – If True, NaNs are considered equal. +

      • equal_nan (bool) – If True, NaNs are considered equal. Defaults to False.

      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 1844f57ef..3089ef645 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.29.2 documentation + mlx.core.as_strided — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.as_strided#

      -as_strided(a: array, /, shape: Sequence[int] | None = None, strides: Sequence[int] | None = None, offset: int = 0, *, stream: None | Stream | Device = None) → array#
      +as_strided(a: array, /, shape: Sequence[int] | None = None, strides: Sequence[int] | None = None, offset: int = 0, *, stream: None | Stream | Device = None) → array#

      Create a view into the array with the given shape and strides.

      The resulting array will always be as if the provided array was row contiguous regardless of the provided arrays storage order and current @@ -948,12 +948,12 @@ result into crashes.

      Parameters:

      • a (array) – Input array

        • -

      • shape (list(int), optional) – The shape of the resulting array. If +

      • shape (list(int), optional) – The shape of the resulting array. If None it defaults to a.shape().

        • -

      • strides (list(int), optional) – The strides of the resulting array. If +

      • strides (list(int), optional) – The strides of the resulting array. If None it defaults to the reverse exclusive cumulative product of a.shape().

        • -

      • offset (int) – Skip that many elements from the beginning of the input +

      • offset (int) – Skip that many elements from the beginning of the input array.

      diff --git a/docs/build/html/python/_autosummary/mlx.core.async_eval.html b/docs/build/html/python/_autosummary/mlx.core.async_eval.html index bc64ac004..477a97243 100644 --- a/docs/build/html/python/_autosummary/mlx.core.async_eval.html +++ b/docs/build/html/python/_autosummary/mlx.core.async_eval.html @@ -8,7 +8,7 @@ - mlx.core.async_eval — MLX 0.29.2 documentation + mlx.core.async_eval — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -942,7 +942,7 @@ document.write(`
      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 +list, tuple or dict. Leaves which are not arrays are ignored.

      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 2bef3d7f8..46822c367 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.29.2 documentation + mlx.core.atleast_1d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.atleast_1d#

      -atleast_1d(*arys: array, stream: None | Stream | Device = None) → array | list[array]#
      +atleast_1d(*arys: array, stream: None | Stream | Device = None) → array | list[array]#

      Convert all arrays to have at least one dimension.

      Parameters:
      @@ -945,7 +945,7 @@ document.write(`

      An array or list of arrays with at least one dimension.

      Return type:
      -

      array or list(array)

      +

      array or list(array)

      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 dc9b3f14a..e463c575d 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.29.2 documentation + mlx.core.atleast_2d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.atleast_2d#

      -atleast_2d(*arys: array, stream: None | Stream | Device = None) → array | list[array]#
      +atleast_2d(*arys: array, stream: None | Stream | Device = None) → array | list[array]#

      Convert all arrays to have at least two dimensions.

      Parameters:
      @@ -945,7 +945,7 @@ document.write(`

      An array or list of arrays with at least two dimensions.

      Return type:
      -

      array or list(array)

      +

      array or list(array)

      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 b4187107e..490694b7d 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.29.2 documentation + mlx.core.atleast_3d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.atleast_3d#

      -atleast_3d(*arys: array, stream: None | Stream | Device = None) → array | list[array]#
      +atleast_3d(*arys: array, stream: None | Stream | Device = None) → array | list[array]#

      Convert all arrays to have at least three dimensions.

      Parameters:
      @@ -945,7 +945,7 @@ document.write(`

      An array or list of arrays with at least three dimensions.

      Return type:
      -

      array or list(array)

      +

      array or list(array)

      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 3483adc14..01e32834a 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.29.2 documentation + mlx.core.bitwise_and — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.bitwise_and#

      -bitwise_and(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
      +bitwise_and(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

      Element-wise bitwise and.

      Take the bitwise and of two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

      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 f0d448581..c1769ad7e 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.29.2 documentation + mlx.core.bitwise_invert — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.bitwise_invert#

      -bitwise_invert(a: scalar | array, stream: None | Stream | Device = None) → array#
      +bitwise_invert(a: scalar | array, stream: None | Stream | Device = None) → array#

      Element-wise bitwise inverse.

      Take the bitwise complement of the input.

      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 a612f3c01..a3b02c1e5 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.29.2 documentation + mlx.core.bitwise_or — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.bitwise_or#

      -bitwise_or(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
      +bitwise_or(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

      Element-wise bitwise or.

      Take the bitwise or of two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

      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 6f823f4c7..5e542dcfb 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.29.2 documentation + mlx.core.bitwise_xor — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

      mlx.core.bitwise_xor#

      -bitwise_xor(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
      +bitwise_xor(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

      Element-wise bitwise xor.

      Take the bitwise exclusive or of two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be 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 d25ac436f..cad137f2f 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.29.2 documentation + mlx.core.block_masked_mm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

      mlx.core.block_masked_mm#

      -block_masked_mm(a: array, b: array, /, block_size: int = 64, mask_out: array | None = None, mask_lhs: array | None = None, mask_rhs: array | None = None, *, stream: None | Stream | Device = None) → array#
      +block_masked_mm(a: array, b: array, /, block_size: int = 64, mask_out: array | None = None, mask_lhs: array | None = None, mask_rhs: array | None = None, *, stream: None | Stream | Device = None) → array#

      Matrix multiplication with block masking.

      Perform the (possibly batched) matrix multiplication of two arrays and with blocks of size block_size x block_size optionally masked out.

      @@ -949,7 +949,7 @@ of size block_size<

      • a (array) – Input array or scalar.

      • b (array) – Input array or scalar.

        • -

      • block_size (int) – Size of blocks to be masked. Must be 32 or 64. Default: 64.

        • +

      • block_size (int) – Size of blocks to be masked. Must be 32 or 64. Default: 64.

      • mask_out (array, optional) – Mask for output. Default: None.

      • mask_lhs (array, optional) – Mask for a. Default: None.

      • mask_rhs (array, optional) – Mask for b. Default: None.

        • diff --git a/docs/build/html/python/_autosummary/mlx.core.broadcast_arrays.html b/docs/build/html/python/_autosummary/mlx.core.broadcast_arrays.html index 73d0d3e6e..a6d233cd7 100644 --- a/docs/build/html/python/_autosummary/mlx.core.broadcast_arrays.html +++ b/docs/build/html/python/_autosummary/mlx.core.broadcast_arrays.html @@ -8,7 +8,7 @@ - mlx.core.broadcast_arrays — MLX 0.29.2 documentation + mlx.core.broadcast_arrays — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.broadcast_arrays#

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

        Broadcast arrays against one another.

        The broadcasting semantics are the same as Numpy.

        @@ -943,7 +943,7 @@ document.write(`

        The output arrays with the broadcasted shape.

        Return type:
        -

        tuple(array)

        +

        tuple(array)

        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 4e5452991..4e18a8524 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.29.2 documentation + mlx.core.broadcast_to — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

        mlx.core.broadcast_to#

        -broadcast_to(a: scalar | array, /, shape: Sequence[int], *, stream: None | Stream | Device = None) → array#
        +broadcast_to(a: scalar | array, /, shape: Sequence[int], *, stream: None | Stream | Device = None) → array#

        Broadcast an array to the given shape.

        The broadcasting semantics are the same as Numpy.

        Parameters:

        • a (array) – Input array.

          • -

        • shape (list(int)) – The shape to broadcast to.

          • +

        • shape (list(int)) – The shape to broadcast to.

        Returns:
        diff --git a/docs/build/html/python/_autosummary/mlx.core.ceil.html b/docs/build/html/python/_autosummary/mlx.core.ceil.html index 3e6d61fd2..51f7d0820 100644 --- a/docs/build/html/python/_autosummary/mlx.core.ceil.html +++ b/docs/build/html/python/_autosummary/mlx.core.ceil.html @@ -8,7 +8,7 @@ - mlx.core.ceil — MLX 0.29.2 documentation + mlx.core.ceil — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.ceil#

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

        Element-wise ceil.

        Parameters:
        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 1c180f8cf..0faa44f69 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.29.2 documentation + mlx.core.clear_cache — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.clear_cache#

        -clear_cache() → None#
        +clear_cache() → None#

        Clear the memory cache.

        After calling this, get_cache_memory() should return 0.

        diff --git a/docs/build/html/python/_autosummary/mlx.core.clip.html b/docs/build/html/python/_autosummary/mlx.core.clip.html index 9722b27b4..1853762c2 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.29.2 documentation + mlx.core.clip — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.clip#

        -clip(a: array, /, a_min: scalar | array | None, a_max: scalar | array | None, *, stream: None | Stream | Device = None) → array#
        +clip(a: array, /, a_min: scalar | array | None, a_max: scalar | array | None, *, stream: None | Stream | Device = None) → array#

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

        If either a_min or a_max are None, then corresponding edge is ignored. At least one of a_min and a_max cannot be None. diff --git a/docs/build/html/python/_autosummary/mlx.core.compile.html b/docs/build/html/python/_autosummary/mlx.core.compile.html index bc04ef254..a30dd5bc8 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.29.2 documentation + mlx.core.compile — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.compile#

        -compile(fun: Callable, inputs: object | None = None, outputs: object | None = None, shapeless: bool = False) → Callable#
        +compile(fun: Callable, inputs: object | None = None, outputs: object | None = None, shapeless: bool = False) → Callable#

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

        Parameters:
        @@ -940,17 +940,17 @@ document.write(`

      • fun (Callable) – A function which takes a variable number of array or trees of array and returns a variable number of array or trees of array.

        • -

      • inputs (list or dict, optional) – These inputs will be captured during +

      • inputs (list or dict, optional) – These inputs will be captured during the function compilation along with the inputs to fun. The inputs -can be a list or a dict containing arbitrarily nested +can be a list or a dict containing arbitrarily nested lists, dictionaries, or arrays. Leaf nodes that are not array are ignored. Default: None

        • -

      • outputs (list or dict, optional) – These outputs will be captured and +

      • outputs (list or dict, optional) – These outputs will be captured and updated in a compiled function. The outputs can be a -list or a dict containing arbitrarily nested lists, +list or a dict containing arbitrarily nested lists, dictionaries, or arrays. Leaf nodes that are not array are ignored. Default: None

        • -

      • shapeless (bool, optional) – A function compiled with the shapeless +

      • shapeless (bool, optional) – A function compiled with the shapeless option enabled will not be recompiled when the input shape changes. Not all functions can be compiled with shapeless enabled. Attempting to compile such functions with shapeless enabled will throw. Note, changing the number diff --git a/docs/build/html/python/_autosummary/mlx.core.concatenate.html b/docs/build/html/python/_autosummary/mlx.core.concatenate.html index 653eeea95..4a4cec769 100644 --- a/docs/build/html/python/_autosummary/mlx.core.concatenate.html +++ b/docs/build/html/python/_autosummary/mlx.core.concatenate.html @@ -8,7 +8,7 @@ - mlx.core.concatenate — MLX 0.29.2 documentation + mlx.core.concatenate — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

        mlx.core.concatenate#

        -concatenate(arrays: list[array], axis: int | None = 0, *, stream: None | Stream | Device = None) → array#
        +concatenate(arrays: list[array], axis: int | None = 0, *, stream: None | Stream | Device = None) → array#

        Concatenate the arrays along the given axis.

        Parameters:
          -

        • arrays (list(array)) – Input list or tuple of arrays.

          • -

        • axis (int, optional) – Optional axis to concatenate along. If +

        • arrays (list(array)) – Input list or tuple of arrays.

          • +

        • axis (int, optional) – Optional axis to concatenate along. If unspecified defaults to 0.

        diff --git a/docs/build/html/python/_autosummary/mlx.core.conj.html b/docs/build/html/python/_autosummary/mlx.core.conj.html index 2146a84ec..196f3267b 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.29.2 documentation + mlx.core.conj — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.conj#

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

        Return the elementwise complex conjugate of the input. Alias for mx.conjugate.

        diff --git a/docs/build/html/python/_autosummary/mlx.core.conjugate.html b/docs/build/html/python/_autosummary/mlx.core.conjugate.html index b9df96599..1e1d0f9c0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.conjugate.html +++ b/docs/build/html/python/_autosummary/mlx.core.conjugate.html @@ -8,7 +8,7 @@ - mlx.core.conjugate — MLX 0.29.2 documentation + mlx.core.conjugate — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.conjugate#

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

        Return the elementwise complex conjugate of the input. Alias for mx.conj.

        diff --git a/docs/build/html/python/_autosummary/mlx.core.contiguous.html b/docs/build/html/python/_autosummary/mlx.core.contiguous.html index f9acdeee3..7033cda02 100644 --- a/docs/build/html/python/_autosummary/mlx.core.contiguous.html +++ b/docs/build/html/python/_autosummary/mlx.core.contiguous.html @@ -8,7 +8,7 @@ - mlx.core.contiguous — MLX 0.29.2 documentation + mlx.core.contiguous — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

        mlx.core.contiguous#

        -contiguous(a: array, /, allow_col_major: bool = False, *, stream: None | Stream | Device = None) → array#
        +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

          • +

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

        Returns:
        diff --git a/docs/build/html/python/_autosummary/mlx.core.conv1d.html b/docs/build/html/python/_autosummary/mlx.core.conv1d.html index 04ba47054..a68852892 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.29.2 documentation + mlx.core.conv1d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

        mlx.core.conv1d#

        -conv1d(input: array, weight: array, /, stride: int = 1, padding: int = 0, dilation: int = 1, groups: int = 1, *, stream: None | Stream | Device = None) → array#
        +conv1d(input: array, weight: array, /, stride: int = 1, padding: int = 0, dilation: int = 1, groups: int = 1, *, stream: None | Stream | Device = None) → array#

        1D convolution over an input with several channels

        Parameters:

        • input (array) – Input array of shape (N, L, C_in).

        • weight (array) – Weight array of shape (C_out, K, C_in).

          • -

        • stride (int, optional) – Kernel stride. Default: 1.

          • -

        • padding (int, optional) – Input padding. Default: 0.

          • -

        • dilation (int, optional) – Kernel dilation. Default: 1.

          • -

        • groups (int, optional) – Input feature groups. Default: 1.

          • +

        • stride (int, optional) – Kernel stride. Default: 1.

          • +

        • padding (int, optional) – Input padding. Default: 0.

          • +

        • dilation (int, optional) – Kernel dilation. Default: 1.

          • +

        • groups (int, optional) – Input feature groups. Default: 1.

        Returns:
        diff --git a/docs/build/html/python/_autosummary/mlx.core.conv2d.html b/docs/build/html/python/_autosummary/mlx.core.conv2d.html index 4e1e2e104..1fd0536f1 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.29.2 documentation + mlx.core.conv2d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,23 +932,23 @@ document.write(`

        mlx.core.conv2d#

        -conv2d(input: array, weight: array, /, stride: int | tuple[int, int] = 1, padding: int | tuple[int, int] = 0, dilation: int | tuple[int, int] = 1, groups: int = 1, *, stream: None | Stream | Device = None) → array#
        +conv2d(input: array, weight: array, /, stride: int | tuple[int, int] = 1, padding: int | tuple[int, int] = 0, dilation: int | tuple[int, int] = 1, groups: int = 1, *, stream: None | Stream | Device = None) → array#

        2D convolution over an input with several channels

        Parameters:

        • input (array) – Input array of shape (N, H, W, C_in).

        • weight (array) – Weight array of shape (C_out, KH, KW, C_in).

          • -

        • stride (int or tuple(int), optional) – tuple of size 2 with +

        • stride (int or tuple(int), optional) – tuple of size 2 with kernel strides. All spatial dimensions get the same stride if only one number is specified. Default: 1.

          • -

        • padding (int or tuple(int), optional) – tuple of size 2 with +

        • padding (int or tuple(int), optional) – tuple of size 2 with symmetric input padding. All spatial dimensions get the same padding if only one number is specified. Default: 0.

          • -

        • dilation (int or tuple(int), optional) – tuple of size 2 with +

        • dilation (int or tuple(int), optional) – tuple of size 2 with kernel dilation. All spatial dimensions get the same dilation if only one number is specified. Default: 1

          • -

        • groups (int, optional) – input feature groups. Default: 1.

          • +

        • groups (int, optional) – input feature groups. Default: 1.

        Returns:
        diff --git a/docs/build/html/python/_autosummary/mlx.core.conv3d.html b/docs/build/html/python/_autosummary/mlx.core.conv3d.html index 9bb6bccfa..72ef40086 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.29.2 documentation + mlx.core.conv3d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.conv3d#

        -conv3d(input: array, weight: array, /, stride: int | tuple[int, int, int] = 1, padding: int | tuple[int, int, int] = 0, dilation: int | tuple[int, int, int] = 1, groups: int = 1, *, stream: None | Stream | Device = None) → array#
        +conv3d(input: array, weight: array, /, stride: int | tuple[int, int, int] = 1, padding: int | tuple[int, int, int] = 0, dilation: int | tuple[int, int, int] = 1, groups: int = 1, *, stream: None | Stream | Device = None) → array#

        3D convolution over an input with several channels

        Note: Only the default groups=1 is currently supported.

        @@ -940,16 +940,16 @@ document.write(`

        • input (array) – Input array of shape (N, D, H, W, C_in).

        • weight (array) – Weight array of shape (C_out, KD, KH, KW, C_in).

          • -

        • stride (int or tuple(int), optional) – tuple of size 3 with +

        • stride (int or tuple(int), optional) – tuple of size 3 with kernel strides. All spatial dimensions get the same stride if only one number is specified. Default: 1.

          • -

        • padding (int or tuple(int), optional) – tuple of size 3 with +

        • padding (int or tuple(int), optional) – tuple of size 3 with symmetric input padding. All spatial dimensions get the same padding if only one number is specified. Default: 0.

          • -

        • dilation (int or tuple(int), optional) – tuple of size 3 with +

        • dilation (int or tuple(int), optional) – tuple of size 3 with kernel dilation. All spatial dimensions get the same dilation if only one number is specified. Default: 1

          • -

        • groups (int, optional) – input feature groups. Default: 1.

          • +

        • groups (int, optional) – input feature groups. Default: 1.

        Returns:
        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 001509b97..d601ce0ed 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.29.2 documentation + mlx.core.conv_general — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,26 +932,26 @@ document.write(`

        mlx.core.conv_general#

        -conv_general(input: array, weight: array, /, stride: int | Sequence[int] = 1, padding: int | Sequence[int] | tuple[Sequence[int], Sequence[int]] = 0, kernel_dilation: int | Sequence[int] = 1, input_dilation: int | Sequence[int] = 1, groups: int = 1, flip: bool = False, *, stream: None | Stream | Device = None) → array#
        +conv_general(input: array, weight: array, /, stride: int | Sequence[int] = 1, padding: int | Sequence[int] | tuple[Sequence[int], Sequence[int]] = 0, kernel_dilation: int | Sequence[int] = 1, input_dilation: int | Sequence[int] = 1, groups: int = 1, flip: bool = False, *, stream: None | Stream | Device = None) → array#

        General convolution over an input with several channels

        Parameters:

        • input (array) – Input array of shape (N, ..., C_in).

        • weight (array) – Weight array of shape (C_out, ..., C_in).

          • -

        • stride (int or list(int), optional) – list with kernel strides. +

        • stride (int or list(int), optional) – list with kernel strides. All spatial dimensions get the same stride if only one number is specified. Default: 1.

          • -

        • padding (int, list(int), or tuple(list(int), list(int)), optional) – list with input padding. All spatial dimensions get the same +

        • padding (int, list(int), or tuple(list(int), list(int)), optional) – list with input padding. All spatial dimensions get the same padding if only one number is specified. Default: 0.

          • -

        • kernel_dilation (int or list(int), optional) – list with +

        • kernel_dilation (int or list(int), optional) – list with kernel dilation. All spatial dimensions get the same dilation if only one number is specified. Default: 1

          • -

        • input_dilation (int or list(int), optional) – list with +

        • input_dilation (int or list(int), optional) – list with input dilation. All spatial dimensions get the same dilation if only one number is specified. Default: 1

          • -

        • groups (int, optional) – Input feature groups. Default: 1.

          • -

        • flip (bool, optional) – Flip the order in which the spatial dimensions of +

        • groups (int, optional) – Input feature groups. Default: 1.

          • +

        • flip (bool, optional) – Flip the order in which the spatial dimensions of the weights are processed. Performs the cross-correlation operator when flip is False and the convolution operator otherwise. Default: False.

          • 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 80225ef6e..9574eded6 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.29.2 documentation + mlx.core.conv_transpose1d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,18 +932,18 @@ document.write(`

          mlx.core.conv_transpose1d#

          -conv_transpose1d(input: array, weight: array, /, stride: int = 1, padding: int = 0, dilation: int = 1, output_padding: int = 0, groups: int = 1, *, stream: None | Stream | Device = None) → array#
          +conv_transpose1d(input: array, weight: array, /, stride: int = 1, padding: int = 0, dilation: int = 1, output_padding: int = 0, groups: int = 1, *, stream: None | Stream | Device = None) → array#

          1D transposed convolution over an input with several channels

          Parameters:

          • input (array) – Input array of shape (N, L, C_in).

          • weight (array) – Weight array of shape (C_out, K, C_in).

            • -

          • stride (int, optional) – Kernel stride. Default: 1.

            • -

          • padding (int, optional) – Input padding. Default: 0.

            • -

          • dilation (int, optional) – Kernel dilation. Default: 1.

            • -

          • output_padding (int, optional) – Output padding. Default: 0.

            • -

          • groups (int, optional) – Input feature groups. Default: 1.

            • +

          • stride (int, optional) – Kernel stride. Default: 1.

            • +

          • padding (int, optional) – Input padding. Default: 0.

            • +

          • dilation (int, optional) – Kernel dilation. Default: 1.

            • +

          • output_padding (int, optional) – Output padding. Default: 0.

            • +

          • groups (int, optional) – Input feature groups. Default: 1.

          Returns:
          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 73b7c335c..aeb23f841 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.29.2 documentation + mlx.core.conv_transpose2d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

          mlx.core.conv_transpose2d#

          -conv_transpose2d(input: array, weight: array, /, stride: int | Tuple[int, int] = 1, padding: int | Tuple[int, int] = 0, dilation: int | Tuple[int, int] = 1, output_padding: int | Tuple[int, int] = 0, groups: int = 1, *, stream: None | Stream | Device = None) → array#
          +conv_transpose2d(input: array, weight: array, /, stride: int | Tuple[int, int] = 1, padding: int | Tuple[int, int] = 0, dilation: int | Tuple[int, int] = 1, output_padding: int | Tuple[int, int] = 0, groups: int = 1, *, stream: None | Stream | Device = None) → array#

          2D transposed convolution over an input with several channels

          Note: Only the default groups=1 is currently supported.

          @@ -940,19 +940,19 @@ document.write(`

          • input (array) – Input array of shape (N, H, W, C_in).

          • weight (array) – Weight array of shape (C_out, KH, KW, C_in).

            • -

          • stride (int or tuple(int), optional) – tuple of size 2 with +

          • stride (int or tuple(int), optional) – tuple of size 2 with kernel strides. All spatial dimensions get the same stride if only one number is specified. Default: 1.

            • -

          • padding (int or tuple(int), optional) – tuple of size 2 with +

          • padding (int or tuple(int), optional) – tuple of size 2 with symmetric input padding. All spatial dimensions get the same padding if only one number is specified. Default: 0.

            • -

          • dilation (int or tuple(int), optional) – tuple of size 2 with +

          • dilation (int or tuple(int), optional) – tuple of size 2 with kernel dilation. All spatial dimensions get the same dilation if only one number is specified. Default: 1

            • -

          • output_padding (int or tuple(int), optional) – tuple of size 2 with +

          • output_padding (int or tuple(int), optional) – tuple of size 2 with output padding. All spatial dimensions get the same output padding if only one number is specified. Default: 0.

            • -

          • groups (int, optional) – input feature groups. Default: 1.

            • +

          • groups (int, optional) – input feature groups. Default: 1.

          Returns:
          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 79174d458..9934a3374 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.29.2 documentation + mlx.core.conv_transpose3d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

          mlx.core.conv_transpose3d#

          -conv_transpose3d(input: array, weight: array, /, stride: int | Tuple[int, int, int] = 1, padding: int | Tuple[int, int, int] = 0, dilation: int | Tuple[int, int, int] = 1, output_padding: int | Tuple[int, int, int] = 0, groups: int = 1, *, stream: None | Stream | Device = None) → array#
          +conv_transpose3d(input: array, weight: array, /, stride: int | Tuple[int, int, int] = 1, padding: int | Tuple[int, int, int] = 0, dilation: int | Tuple[int, int, int] = 1, output_padding: int | Tuple[int, int, int] = 0, groups: int = 1, *, stream: None | Stream | Device = None) → array#

          3D transposed convolution over an input with several channels

          Note: Only the default groups=1 is currently supported.

          @@ -940,19 +940,19 @@ document.write(`

          • input (array) – Input array of shape (N, D, H, W, C_in).

          • weight (array) – Weight array of shape (C_out, KD, KH, KW, C_in).

            • -

          • stride (int or tuple(int), optional) – tuple of size 3 with +

          • stride (int or tuple(int), optional) – tuple of size 3 with kernel strides. All spatial dimensions get the same stride if only one number is specified. Default: 1.

            • -

          • padding (int or tuple(int), optional) – tuple of size 3 with +

          • padding (int or tuple(int), optional) – tuple of size 3 with symmetric input padding. All spatial dimensions get the same padding if only one number is specified. Default: 0.

            • -

          • dilation (int or tuple(int), optional) – tuple of size 3 with +

          • dilation (int or tuple(int), optional) – tuple of size 3 with kernel dilation. All spatial dimensions get the same dilation if only one number is specified. Default: 1

            • -

          • output_padding (int or tuple(int), optional) – tuple of size 3 with +

          • output_padding (int or tuple(int), optional) – tuple of size 3 with output padding. All spatial dimensions get the same output padding if only one number is specified. Default: 0.

            • -

          • groups (int, optional) – input feature groups. Default: 1.

            • +

          • groups (int, optional) – input feature groups. Default: 1.

          Returns:
          diff --git a/docs/build/html/python/_autosummary/mlx.core.convolve.html b/docs/build/html/python/_autosummary/mlx.core.convolve.html index 60ac75072..5ea3adf83 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.29.2 documentation + mlx.core.convolve — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

          mlx.core.convolve#

          -convolve(a: array, v: array, /, mode: str = 'full', *, stream: None | Stream | Device = None) → array#
          +convolve(a: array, v: array, /, mode: str = 'full', *, stream: None | Stream | Device = None) → array#

          The discrete convolution of 1D arrays.

          If v is longer than a, then they are swapped. The conv filter is flipped following signal processing convention.

          @@ -941,7 +941,7 @@ The conv filter is flipped following signal processing convention.

          • a (array) – 1D Input array.

          • v (array) – 1D Input array.

            • -

          • mode (str, optional) – {‘full’, ‘valid’, ‘same’}

            • +

          • mode (str, optional) – {‘full’, ‘valid’, ‘same’}

          Returns:
          diff --git a/docs/build/html/python/_autosummary/mlx.core.cos.html b/docs/build/html/python/_autosummary/mlx.core.cos.html index f39af5ea4..d5af5c768 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.29.2 documentation + mlx.core.cos — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

          mlx.core.cos#

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

          Element-wise cosine.

          Parameters:
          diff --git a/docs/build/html/python/_autosummary/mlx.core.cosh.html b/docs/build/html/python/_autosummary/mlx.core.cosh.html index 6e0626d83..fe5425201 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.29.2 documentation + mlx.core.cosh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

          mlx.core.cosh#

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

          Element-wise hyperbolic cosine.

          Parameters:
          diff --git a/docs/build/html/python/_autosummary/mlx.core.cuda.is_available.html b/docs/build/html/python/_autosummary/mlx.core.cuda.is_available.html index 15b84273b..e7af101de 100644 --- a/docs/build/html/python/_autosummary/mlx.core.cuda.is_available.html +++ b/docs/build/html/python/_autosummary/mlx.core.cuda.is_available.html @@ -8,7 +8,7 @@ - mlx.core.cuda.is_available — MLX 0.29.2 documentation + mlx.core.cuda.is_available — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

          mlx.core.cuda.is_available#

          -is_available() → bool#
          +is_available() → bool#

          Check if the CUDA back-end is available.

          diff --git a/docs/build/html/python/_autosummary/mlx.core.cummax.html b/docs/build/html/python/_autosummary/mlx.core.cummax.html index 66904c6ee..ca3c63884 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.29.2 documentation + mlx.core.cummax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

          mlx.core.cummax#

          -cummax(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#
          +cummax(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#

          Return the cumulative maximum of the elements along the given axis.

          Parameters:

          • a (array) – Input array

            • -

          • axis (int, optional) – Optional axis to compute the cumulative maximum +

          • axis (int, optional) – Optional axis to compute the cumulative maximum over. If unspecified the cumulative maximum of the flattened array is returned.

            • -

          • reverse (bool) – Perform the cumulative maximum in reverse.

            • -

          • inclusive (bool) – The i-th element of the output includes the i-th +

          • reverse (bool) – Perform the cumulative maximum in reverse.

            • +

          • inclusive (bool) – The i-th element of the output includes the i-th element of the input.

          diff --git a/docs/build/html/python/_autosummary/mlx.core.cummin.html b/docs/build/html/python/_autosummary/mlx.core.cummin.html index 1ed73da00..b8ef6c806 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.29.2 documentation + mlx.core.cummin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

          mlx.core.cummin#

          -cummin(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#
          +cummin(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#

          Return the cumulative minimum of the elements along the given axis.

          Parameters:

          • a (array) – Input array

            • -

          • axis (int, optional) – Optional axis to compute the cumulative minimum +

          • axis (int, optional) – Optional axis to compute the cumulative minimum over. If unspecified the cumulative minimum of the flattened array is returned.

            • -

          • reverse (bool) – Perform the cumulative minimum in reverse.

            • -

          • inclusive (bool) – The i-th element of the output includes the i-th +

          • reverse (bool) – Perform the cumulative minimum in reverse.

            • +

          • inclusive (bool) – The i-th element of the output includes the i-th element of the input.

          diff --git a/docs/build/html/python/_autosummary/mlx.core.cumprod.html b/docs/build/html/python/_autosummary/mlx.core.cumprod.html index fb81b6a13..5250357eb 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.29.2 documentation + mlx.core.cumprod — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

          mlx.core.cumprod#

          -cumprod(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#
          +cumprod(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#

          Return the cumulative product of the elements along the given axis.

          Parameters:

          • a (array) – Input array

            • -

          • axis (int, optional) – Optional axis to compute the cumulative product +

          • axis (int, optional) – Optional axis to compute the cumulative product over. If unspecified the cumulative product of the flattened array is returned.

            • -

          • reverse (bool) – Perform the cumulative product in reverse.

            • -

          • inclusive (bool) – The i-th element of the output includes the i-th +

          • reverse (bool) – Perform the cumulative product in reverse.

            • +

          • inclusive (bool) – The i-th element of the output includes the i-th element of the input.

          diff --git a/docs/build/html/python/_autosummary/mlx.core.cumsum.html b/docs/build/html/python/_autosummary/mlx.core.cumsum.html index 87e1bc8fe..5030d4204 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.29.2 documentation + mlx.core.cumsum — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

          mlx.core.cumsum#

          -cumsum(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#
          +cumsum(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#

          Return the cumulative sum of the elements along the given axis.

          Parameters:

          • a (array) – Input array

            • -

          • axis (int, optional) – Optional axis to compute the cumulative sum +

          • axis (int, optional) – Optional axis to compute the cumulative sum over. If unspecified the cumulative sum of the flattened array is returned.

            • -

          • reverse (bool) – Perform the cumulative sum in reverse.

            • -

          • inclusive (bool) – The i-th element of the output includes the i-th +

          • reverse (bool) – Perform the cumulative sum in reverse.

            • +

          • inclusive (bool) – The i-th element of the output includes the i-th element of the input.

          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 7a73b4639..0db2a36b4 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.29.2 documentation + mlx.core.custom_function — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 54afcafce..1812d41d6 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.29.2 documentation + mlx.core.default_device — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 333cba83e..73e0c1e2e 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.29.2 documentation + mlx.core.default_stream — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/_autosummary/mlx.core.degrees.html b/docs/build/html/python/_autosummary/mlx.core.degrees.html index 3909cc1c4..8fd0da3ff 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.29.2 documentation + mlx.core.degrees — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

          mlx.core.degrees#

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

          Convert angles from radians to degrees.

          Parameters:
          diff --git a/docs/build/html/python/_autosummary/mlx.core.dequantize.html b/docs/build/html/python/_autosummary/mlx.core.dequantize.html index af3d727ad..36a00fa7d 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.29.2 documentation + mlx.core.dequantize — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

          mlx.core.dequantize#

          -dequantize(w: array, /, scales: array, biases: array | None = None, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, stream: None | Stream | Device = None) → array#
          +dequantize(w: array, /, scales: array, biases: array | None = None, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, stream: None | Stream | Device = None) → array#

          Dequantize the matrix w using quantization parameters.

          Parameters:
          @@ -942,11 +942,11 @@ document.write(`

        • scales (array) – The scales to use per group_size elements of w.

        • biases (array, optional) – The biases to use per group_size elements of w. Default: None.

          • -

        • group_size (int, optional) – The size of the group in w that shares a +

        • group_size (int, optional) – The size of the group in w that shares a scale and bias. Default: 64.

          • -

        • bits (int, optional) – The number of bits occupied by each element in +

        • bits (int, optional) – The number of bits occupied by each element in w. Default: 4.

          • -

        • mode (str, optional) – The quantization mode. Default: "affine".

          • +

        • mode (str, optional) – The quantization mode. Default: "affine".

        Returns:
        diff --git a/docs/build/html/python/_autosummary/mlx.core.diag.html b/docs/build/html/python/_autosummary/mlx.core.diag.html index a5b7539c1..2bf766d95 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.29.2 documentation + mlx.core.diag — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

        mlx.core.diag#

        -diag(a: array, /, k: int = 0, *, stream: None | Stream | Device = None) → array#
        +diag(a: array, /, k: int = 0, *, stream: None | Stream | Device = None) → array#

        Extract a diagonal or construct a diagonal matrix. If a is 1-D then a diagonal matrix is constructed with a on the \(k\)-th diagonal. If a is 2-D then the \(k\)-th diagonal is @@ -942,7 +942,7 @@ returned.

        Parameters:

        • a (array) – 1-D or 2-D input array.

          • -

        • k (int, optional) – The diagonal to extract or construct. +

        • k (int, optional) – The diagonal to extract or construct. Default: 0.

        diff --git a/docs/build/html/python/_autosummary/mlx.core.diagonal.html b/docs/build/html/python/_autosummary/mlx.core.diagonal.html index b0fb55425..c1be958f8 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.29.2 documentation + mlx.core.diagonal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.diagonal#

        -diagonal(a: array, offset: int = 0, axis1: int = 0, axis2: int = 1, stream: None | Stream | Device = None) → array#
        +diagonal(a: array, offset: int = 0, axis1: int = 0, axis2: int = 1, stream: None | Stream | Device = None) → array#

        Return specified diagonals.

        If a is 2-D, then a 1-D array containing the diagonal at the given offset is returned.

        @@ -944,11 +944,11 @@ new dimension inserted at the end corresponding to the diagonal.

        Parameters:

        • a (array) – Input array

          • -

        • offset (int, optional) – Offset of the diagonal from the main diagonal. +

        • offset (int, optional) – Offset of the diagonal from the main diagonal. Can be positive or negative. Default: 0.

          • -

        • axis1 (int, optional) – The first axis of the 2-D sub-arrays from which +

        • axis1 (int, optional) – The first axis of the 2-D sub-arrays from which the diagonals should be taken. Default: 0.

          • -

        • axis2 (int, optional) – The second axis of the 2-D sub-arrays from which +

        • axis2 (int, optional) – The second axis of the 2-D sub-arrays from which the diagonals should be taken. Default: 1.

        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 859056597..a27149c64 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.29.2 documentation + mlx.core.disable_compile — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.disable_compile#

        -disable_compile() → None#
        +disable_compile() → None#

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

        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 36acc55f9..cd91d103c 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.29.2 documentation + mlx.core.distributed.Group — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 23947803a..dfd6a39c8 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.29.2 documentation + mlx.core.distributed.all_gather — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.distributed.all_gather#

        -all_gather(x: array, *, group: Group | None = None, stream: None | Stream | Device = None) → array#
        +all_gather(x: array, *, group: Group | None = None, stream: None | Stream | Device = None) → array#

        Gather arrays from all processes.

        Gather the x arrays from all processes in the group and concatenate them along the first axis. The arrays should all have the same shape.

        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 b9b3be5b7..05a392de6 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.29.2 documentation + mlx.core.distributed.all_sum — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.distributed.all_sum#

        -all_sum(x: array, *, group: Group | None = None, stream: None | Stream | Device = None) → array#
        +all_sum(x: array, *, group: Group | None = None, stream: None | Stream | Device = None) → array#

        All reduce sum.

        Sum the x arrays from all processes in the group.

        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 5df1b86c1..aed107475 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.29.2 documentation + mlx.core.distributed.init — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

        mlx.core.distributed.init#

        -init(strict: bool = False, backend: str = 'any') → Group#
        +init(strict: bool = False, backend: str = 'any') → Group#

        Initialize the communication backend and create the global communication group.

        Example

        import mlx.core as mx
@@ -943,10 +943,10 @@ document.write(`
 
        
 
        Parameters:
        
 
          
-
        • strict (bool, optional) – If set to False it returns a singleton group
+
        • strict (bool, optional) – If set to False it returns a singleton group
 in case mx.distributed.is_available() returns False otherwise
 it throws a runtime error. Default: False
          • 
-
        • backend (str, optional) – Which distributed backend to initialize.
+
        • backend (str, optional) – Which distributed backend to initialize.
 Possible values mpi, ring, nccl, any. If set to any all
 available backends are tried and the first one that succeeds
 becomes the global group which will be returned in subsequent
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 d00bbf3d8..80880effa 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.29.2 documentation
+    mlx.core.distributed.is_available — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
        @@ -932,7 +932,7 @@ document.write(`

        mlx.core.distributed.is_available#

        -is_available() → bool#
        +is_available() → bool#

        Check if a communication backend is available.

        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 e4e334e33..d95b850c7 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.29.2 documentation + mlx.core.distributed.recv — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
        @@ -932,15 +932,15 @@ document.write(`

        mlx.core.distributed.recv#

        -recv(shape: Sequence[int], dtype: Dtype, src: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array#
        +recv(shape: Sequence[int], dtype: Dtype, src: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array#

        Recv an array with shape shape and dtype dtype from process with rank src.

        Parameters:
          -

        • shape (Tuple[int]) – The shape of the array we are receiving.

          • +

        • shape (Tuple[int]) – The shape of the array we are receiving.

        • dtype (Dtype) – The data type of the array we are receiving.

          • -

        • src (int) – Rank of the source process in the group.

          • +

        • src (int) – Rank of the source process in the group.

        • group (Group) – The group of processes that will participate in the recv. If set to None the global group is used. Default: None.

          • 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 13645ee75..265a8241c 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.29.2 documentation + mlx.core.distributed.recv_like — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

          mlx.core.distributed.recv_like#

          -recv_like(x: array, src: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array#
          +recv_like(x: array, src: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array#

          Recv an array with shape and type like x from process with rank src.

          It is equivalent to calling mx.distributed.recv(x.shape, x.dtype, src).

          @@ -941,7 +941,7 @@ document.write(`

          • x (array) – An array defining the shape and dtype of the array we are receiving.

            • -

          • src (int) – Rank of the source process in the group.

            • +

          • src (int) – Rank of the source process in the group.

          • group (Group) – The group of processes that will participate in the recv. If set to None the global group is used. Default: None.

            • 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 42f8e8866..d7f0403c9 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.29.2 documentation + mlx.core.distributed.send — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

            mlx.core.distributed.send#

            -send(x: array, dst: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array#
            +send(x: array, dst: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array#

            Send an array from the current process to the process that has rank dst in the group.

            Parameters:

            • x (array) – Input array.

              • -

            • dst (int) – Rank of the destination process in the group.

              • +

            • dst (int) – Rank of the destination process in the group.

            • group (Group) – The group of processes that will participate in the sned. If set to None the global group is used. Default: None.

              • diff --git a/docs/build/html/python/_autosummary/mlx.core.divide.html b/docs/build/html/python/_autosummary/mlx.core.divide.html index 92fe4f728..2ead715b4 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.29.2 documentation + mlx.core.divide — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.divide#

              -divide(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
              +divide(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

              Element-wise division.

              Divide two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

              diff --git a/docs/build/html/python/_autosummary/mlx.core.divmod.html b/docs/build/html/python/_autosummary/mlx.core.divmod.html index e818120e2..cd2f0cb55 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.29.2 documentation + mlx.core.divmod — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.divmod#

              -divmod(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
              +divmod(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

              Element-wise quotient and remainder.

              The fuction divmod(a, b) is equivalent to but faster than (a // b, a % b). The function uses numpy-style broadcasting @@ -948,7 +948,7 @@ semantics. Either or both input arrays can also be scalars.

              The quotient a // b and remainder a % b.

              Return type:
              -

              tuple(array, array)

              +

              tuple(array, array)

              diff --git a/docs/build/html/python/_autosummary/mlx.core.einsum.html b/docs/build/html/python/_autosummary/mlx.core.einsum.html index d5abd2dac..e6be38dd4 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.29.2 documentation + mlx.core.einsum — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,12 +932,12 @@ document.write(`

              mlx.core.einsum#

              -einsum(subscripts: str, *operands, stream: None | Stream | Device = None) → array#
              +einsum(subscripts: str, *operands, stream: None | Stream | Device = None) → array#

              Perform the Einstein summation convention on the operands.

              Parameters:
                -

              • subscripts (str) – The Einstein summation convention equation.

                • +

              • subscripts (str) – The Einstein summation convention equation.

              • *operands (array) – The input arrays.

              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 031825fe4..e827916bc 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.29.2 documentation + mlx.core.einsum_path — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,12 +932,12 @@ document.write(`

              mlx.core.einsum_path#

              -einsum_path(subscripts: str, *operands)#
              +einsum_path(subscripts: str, *operands)#

              Compute the contraction order for the given Einstein summation.

              Parameters:
                -

              • subscripts (str) – The Einstein summation convention equation.

                • +

              • subscripts (str) – The Einstein summation convention equation.

              • *operands (array) – The input arrays.

              @@ -946,7 +946,7 @@ document.write(` chosen path.

              Return type:
              -

              tuple(list(tuple(int, int)), str)

              +

              tuple(list(tuple(int, int)), str)

              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 e126bfe0a..6f5c4c4ba 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.29.2 documentation + mlx.core.enable_compile — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.enable_compile#

              -enable_compile() → None#
              +enable_compile() → None#

              Globally enable compilation. This will override the environment variable MLX_DISABLE_COMPILE if set.

              diff --git a/docs/build/html/python/_autosummary/mlx.core.equal.html b/docs/build/html/python/_autosummary/mlx.core.equal.html index 42023fbb2..5977c4191 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.29.2 documentation + mlx.core.equal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.equal#

              -equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
              +equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

              Element-wise equality.

              Equality comparison on two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

              diff --git a/docs/build/html/python/_autosummary/mlx.core.erf.html b/docs/build/html/python/_autosummary/mlx.core.erf.html index 2972dc4b6..eea40c905 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.29.2 documentation + mlx.core.erf — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

              mlx.core.erf#

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

              Element-wise error function.

              \[\mathrm{erf}(x) = \frac{2}{\sqrt{\pi}} \int_0^x e^{-t^2} \, dt\]
              diff --git a/docs/build/html/python/_autosummary/mlx.core.erfinv.html b/docs/build/html/python/_autosummary/mlx.core.erfinv.html index a7509e0d7..e3c9bc0da 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.29.2 documentation + mlx.core.erfinv — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.erfinv#

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

              Element-wise inverse of erf().

              Parameters:
              diff --git a/docs/build/html/python/_autosummary/mlx.core.eval.html b/docs/build/html/python/_autosummary/mlx.core.eval.html index 313b48fa7..7aa207241 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.29.2 documentation + mlx.core.eval — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

              mlx.core.eval#

              -eval(*args) → None#
              +eval(*args) → None#

              Evaluate an array or tree of array.

              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 +list, tuple or dict. Leaves which are not arrays are ignored.

              diff --git a/docs/build/html/python/_autosummary/mlx.core.exp.html b/docs/build/html/python/_autosummary/mlx.core.exp.html index 5e6753e1a..5dea40ff0 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.29.2 documentation + mlx.core.exp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.exp#

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

              Element-wise exponential.

              Parameters:
              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 a566c2807..eebae7964 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.29.2 documentation + mlx.core.expand_dims — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

              mlx.core.expand_dims#

              -expand_dims(a: array, /, axis: int | Sequence[int], *, stream: None | Stream | Device = None) → array#
              +expand_dims(a: array, /, axis: int | Sequence[int], *, stream: None | Stream | Device = None) → array#

              Add a size one dimension at the given axis.

              Parameters:

              • a (array) – Input array.

                • -

              • axes (int or tuple(int)) – The index of the inserted dimensions.

                • +

              • axes (int or tuple(int)) – The index of the inserted dimensions.

              Returns:
              diff --git a/docs/build/html/python/_autosummary/mlx.core.expm1.html b/docs/build/html/python/_autosummary/mlx.core.expm1.html index a5949c292..b67d2c83e 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.29.2 documentation + mlx.core.expm1 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.expm1#

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

              Element-wise exponential minus 1.

              Computes exp(x) - 1 with greater precision for small x.

              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 cbd9dfaa4..5deb2c8af 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.29.2 documentation + mlx.core.export_function — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,8 +932,8 @@ document.write(`

              mlx.core.export_function#

              -export_function(file: str, fun: Callable, *args, shapeless: bool = False, **kwargs) → None#
              -

              Export a function to a file.

              +export_function(arg0: object, fun: Callable, *args, shapeless: bool = False, **kwargs) → None# +

              Export an MLX function.

              Example input arrays must be provided to export a function. The example inputs can be variable *args and **kwargs or a tuple of arrays and/or dictionary of string keys with array values.

              @@ -946,11 +946,12 @@ versions of MLX may not be compatible with future versions.

              Parameters:
                -

              • file (str) – File path to export the function to.

                • +

              • file (str or Callable) – Either a file path to export the function +to or a callback.

              • fun (Callable) – A function which takes as input zero or more array and returns one or more array.

              • *args (array) – Example array inputs to the function.

                • -

              • shapeless (bool, optional) – Whether or not the function allows +

              • shapeless (bool, optional) – Whether or not the function allows inputs with variable shapes. Default: False.

              • **kwargs (array) – Additional example keyword array inputs to the function.

                • 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 07af8976b..3e3fe10dc 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.29.2 documentation + mlx.core.export_to_dot — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                mlx.core.export_to_dot#

                -export_to_dot(file: object, *args, **kwargs) → None#
                +export_to_dot(file: object, *args, **kwargs) → None#

                Export a graph to DOT format for visualization.

                A variable number of output arrays can be provided for exporting The graph exported will recursively include all unevaluated inputs of @@ -940,9 +940,9 @@ the provided outputs.

                Parameters:
                  -

                • file (str) – The file path to export to.

                  • +

                • file (str) – The file path to export to.

                • *args (array) – The output arrays.

                  • -

                • **kwargs (dict[str, array]) – Provide some names for arrays in the +

                • **kwargs (dict[str, array]) – Provide some names for arrays in the graph to make the result easier to parse.

                diff --git a/docs/build/html/python/_autosummary/mlx.core.exporter.html b/docs/build/html/python/_autosummary/mlx.core.exporter.html index 37dd02a2e..ee83ff7e9 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.29.2 documentation + mlx.core.exporter — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                mlx.core.exporter#

                -exporter(file: str, fun: Callable, *, shapeless: bool = False) → mlx.core.FunctionExporter#
                +exporter(file: str, fun: Callable, *, shapeless: bool = False) → mlx.core.FunctionExporter#

                Make a callable object to export multiple traces of a function to a file.

                Warning

                @@ -943,8 +943,8 @@ versions of MLX may not be compatible with future versions.

                Parameters:
                  -

                • file (str) – File path to export the function to.

                  • -

                • shapeless (bool, optional) – Whether or not the function allows +

                • file (str) – File path to export the function to.

                  • +

                • shapeless (bool, optional) – Whether or not the function allows inputs with variable shapes. Default: False.

                diff --git a/docs/build/html/python/_autosummary/mlx.core.eye.html b/docs/build/html/python/_autosummary/mlx.core.eye.html index d7406da85..c608dee47 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.29.2 documentation + mlx.core.eye — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                @@ -932,14 +932,14 @@ document.write(`

                mlx.core.eye#

                -eye(n: int, m: int | None = None, k: int = 0, dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#
                +eye(n: int, m: int | None = None, k: int = 0, dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#

                Create an identity matrix or a general diagonal matrix.

                Parameters:
                  -

                • n (int) – The number of rows in the output.

                  • -

                • m (int, optional) – The number of columns in the output. Defaults to n.

                  • -

                • k (int, optional) – Index of the diagonal. Defaults to 0 (main diagonal).

                  • +

                • n (int) – The number of rows in the output.

                  • +

                • m (int, optional) – The number of columns in the output. Defaults to n.

                  • +

                • k (int, optional) – Index of the diagonal. Defaults to 0 (main diagonal).

                • dtype (Dtype, optional) – Data type of the output array. Defaults to float32.

                • stream (Stream, optional) – Stream or device. Defaults to None.

                diff --git a/docs/build/html/python/_autosummary/mlx.core.fast.cuda_kernel.html b/docs/build/html/python/_autosummary/mlx.core.fast.cuda_kernel.html index a0701309a..c9eb49440 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fast.cuda_kernel.html +++ b/docs/build/html/python/_autosummary/mlx.core.fast.cuda_kernel.html @@ -8,7 +8,7 @@ - mlx.core.fast.cuda_kernel — MLX 0.29.2 documentation + mlx.core.fast.cuda_kernel — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,25 +932,25 @@ document.write(`

                mlx.core.fast.cuda_kernel#

                -cuda_kernel(name: str, input_names: Sequence[str], output_names: Sequence[str], source: str, header: str = '', ensure_row_contiguous: bool = True, shared_memory: int = 0) → object#
                +cuda_kernel(name: str, input_names: Sequence[str], output_names: Sequence[str], source: str, header: str = '', ensure_row_contiguous: bool = True, shared_memory: int = 0) → object#

                A jit-compiled custom CUDA kernel defined from a source string.

                This is the CUDA equivalent of Custom Metal Kernels.

                Parameters:
                  -

                • name (str) – Name for the kernel.

                  • -

                • input_names (List[str]) – The parameter names of the inputs in the +

                • name (str) – Name for the kernel.

                  • +

                • input_names (List[str]) – The parameter names of the inputs in the function signature.

                  • -

                • output_names (List[str]) – The parameter names of the outputs in the +

                • output_names (List[str]) – The parameter names of the outputs in the function signature.

                  • -

                • source (str) – Source code. This is the body of a function in CUDA, +

                • source (str) – Source code. This is the body of a function in CUDA, the function signature will be automatically generated.

                  • -

                • header (str) – Header source code to include before the main function. +

                • header (str) – Header source code to include before the main function. Useful for helper functions or includes that should live outside of the main function body.

                  • -

                • ensure_row_contiguous (bool) – Whether to ensure the inputs are row contiguous +

                • ensure_row_contiguous (bool) – Whether to ensure the inputs are row contiguous before the kernel runs. Default: True.

                  • -

                • shared_memory (int) – The dynamic shared memory to request for the +

                • shared_memory (int) – The dynamic shared memory to request for the kernel. A value of 0 means no dynamic shared memory. Default: 0.

                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 5a062fc4f..6a1c30a7c 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.29.2 documentation + mlx.core.fast.layer_norm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                mlx.core.fast.layer_norm#

                -layer_norm(x: array, weight: array | None, bias: array | None, eps: float, *, stream: None | Stream | Device = None) → array#
                +layer_norm(x: array, weight: array | None, bias: array | None, eps: float, *, stream: None | Stream | Device = None) → array#

                Layer normalization.

                The normalization is with respect to the last axis of the input x.

                @@ -945,7 +945,7 @@ as the last axis of

                bias (array, optional) – An additive offset to be added to the result. The bias should be one-dimensional with the same size as the last axis of x. If set to None then no translation happens.

                -

              • eps (float) – A small additive constant for numerical stability.

                • +

              • eps (float) – A small additive constant for numerical stability.

              Returns:
              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 347bbbe46..9702b1d96 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.29.2 documentation + mlx.core.fast.metal_kernel — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,25 +932,25 @@ document.write(`

              mlx.core.fast.metal_kernel#

              -metal_kernel(name: str, input_names: Sequence[str], output_names: Sequence[str], source: str, header: str = '', ensure_row_contiguous: bool = True, atomic_outputs: bool = False) → object#
              +metal_kernel(name: str, input_names: Sequence[str], output_names: Sequence[str], source: str, header: str = '', ensure_row_contiguous: bool = True, atomic_outputs: bool = False) → object#

              A jit-compiled custom Metal kernel defined from a source string.

              Full documentation: Custom Metal Kernels.

              Parameters:
                -

              • name (str) – Name for the kernel.

                • -

              • input_names (List[str]) – The parameter names of the inputs in the +

              • name (str) – Name for the kernel.

                • +

              • input_names (List[str]) – The parameter names of the inputs in the function signature.

                • -

              • output_names (List[str]) – The parameter names of the outputs in the +

              • output_names (List[str]) – The parameter names of the outputs in the function signature.

                • -

              • source (str) – Source code. This is the body of a function in Metal, +

              • source (str) – Source code. This is the body of a function in Metal, the function signature will be automatically generated.

                • -

              • header (str) – Header source code to include before the main function. +

              • header (str) – Header source code to include before the main function. Useful for helper functions or includes that should live outside of the main function body.

                • -

              • ensure_row_contiguous (bool) – Whether to ensure the inputs are row contiguous +

              • ensure_row_contiguous (bool) – Whether to ensure the inputs are row contiguous before the kernel runs. Default: True.

                • -

              • atomic_outputs (bool) – Whether to use atomic outputs in the function signature +

              • atomic_outputs (bool) – Whether to use atomic outputs in the function signature e.g. device atomic<float>. Default: False.

              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 d444af59a..f5d6d4670 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.29.2 documentation + mlx.core.fast.rms_norm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.fast.rms_norm#

              -rms_norm(x: array, weight: array | None, eps: float, *, stream: None | Stream | Device = None) → array#
              +rms_norm(x: array, weight: array | None, eps: float, *, stream: None | Stream | Device = None) → array#

              Root Mean Square normalization (RMS norm).

              The normalization is with respect to the last axis of the input x.

              @@ -942,7 +942,7 @@ document.write(`

            • weight (array, optional) – A multiplicative weight to scale the result by. The weight should be one-dimensional with the same size as the last axis of x. If set to None then no scaling happens.

              • -

            • eps (float) – A small additive constant for numerical stability.

              • +

            • eps (float) – A small additive constant for numerical stability.

            Returns:
            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 7e77c3271..a58b4bad3 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.29.2 documentation + mlx.core.fast.rope — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.fast.rope#

            -rope(a: array, dims: int, *, traditional: bool, base: float | None, scale: float, offset: int | array, freqs: array | None = None, stream: None | Stream | Device = None) → array#
            +rope(a: array, dims: int, *, traditional: bool, base: float | None, scale: float, offset: int | array, freqs: array | None = None, stream: None | Stream | Device = None) → array#

            Apply rotary positional encoding to the input.

            The input is expected to be at least 3D with shape (B, *, T, D) where:
              @@ -946,16 +946,16 @@ document.write(`
              Parameters:

              • a (array) – The input array.

                • -

              • dims (int) – The feature dimensions to be rotated. If the input feature +

              • dims (int) – The feature dimensions to be rotated. If the input feature is larger than dims then the rest is left unchanged.

                • -

              • traditional (bool) – If set to True choose the traditional +

              • traditional (bool) – If set to True choose the traditional implementation which rotates consecutive dimensions.

                • -

              • base (float, optional) – The base used to compute angular frequency for +

              • base (float, optional) – The base used to compute angular frequency for each dimension in the positional encodings. Exactly one of base and freqs must be None.

                • -

              • scale (float) – The scale used to scale the positions.

                • -

              • offset (int or array) – The position offset to start at. If an -array is given it can be a scalar or vector of B +

              • scale (float) – The scale used to scale the positions.

                • +

              • offset (int or array) – The position offset to start at. If an +array is given it can be a scalar or vector of B offsets for each example in the batch.

              • freqs (array, optional) – Optional frequencies to use with RoPE. If set, the base parameter must be None. Default: None.

                • 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 7f11414bc..403a85acd 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.29.2 documentation + mlx.core.fast.scaled_dot_product_attention — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                mlx.core.fast.scaled_dot_product_attention#

                -scaled_dot_product_attention(q: array, k: array, v: array, *, scale: float, mask: None | str | array = None, sinks: array | None = None, stream: None | Stream | Device = None) → array#
                +scaled_dot_product_attention(q: array, k: array, v: array, *, scale: float, mask: None | str | array = None, sinks: array | None = None, stream: None | Stream | Device = None) → array#

                A fast implementation of multi-head attention: O = softmax(Q @ K.T, dim=-1) @ V.

                Supports:

                  @@ -964,8 +964,8 @@ and v

                • q (array) – Queries with shape [B, N_q, T_q, D].

                • k (array) – Keys with shape [B, N_kv, T_kv, D].

                • v (array) – Values with shape [B, N_kv, T_kv, D].

                  • -

                • scale (float) – Scale for queries (typically 1.0 / sqrt(q.shape(-1)).

                  • -

                • mask (str or array, optional) – The mask to apply to the +

                • scale (float) – Scale for queries (typically 1.0 / sqrt(q.shape(-1)).

                  • +

                • mask (str or array, optional) – The mask to apply to the query-key scores. The mask can be an array or a string indicating the mask type. The only supported string type is "causal". If the mask is an array it can be a boolean or additive mask. The mask 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 f83e73217..7d311aa4f 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.29.2 documentation + mlx.core.fft.fft — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

                  mlx.core.fft.fft#

                  -fft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array#
                  +fft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array#

                  One dimensional discrete Fourier Transform.

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • n (int, optional) – Size of the transformed axis. The +

                  • n (int, optional) – Size of the transformed axis. The corresponding axis in the input is truncated or padded with zeros to match n. The default value is a.shape[axis].

                    • -

                  • axis (int, optional) – Axis along which to perform the FFT. The +

                  • axis (int, optional) – Axis along which to perform the FFT. The default is -1.

                  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 051bf8080..c79b44f85 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.29.2 documentation + mlx.core.fft.fft2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                  mlx.core.fft.fft2#

                  -fft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array#
                  +fft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array#

                  Two dimensional discrete Fourier Transform.

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • s (list(int), optional) – Sizes of the transformed axes. The +

                  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

                    • -

                  • axes (list(int), optional) – Axes along which to perform the FFT. +

                  • axes (list(int), optional) – Axes along which to perform the FFT. The default is [-2, -1].

                  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 1b5ea8bbc..0713a2266 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.29.2 documentation + mlx.core.fft.fftn — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                  mlx.core.fft.fftn#

                  -fftn(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#
                  +fftn(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#

                  n-dimensional discrete Fourier Transform.

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • s (list(int), optional) – Sizes of the transformed axes. The +

                  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

                    • -

                  • axes (list(int), optional) – Axes along which to perform the FFT. +

                  • axes (list(int), optional) – Axes along which to perform the FFT. The default is None in which case the FFT is over the last len(s) axes are or all axes if s is also None.

                  diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.fftshift.html b/docs/build/html/python/_autosummary/mlx.core.fft.fftshift.html index 282e2c6bf..a0aeb40f0 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.fftshift.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.fftshift.html @@ -8,7 +8,7 @@ - mlx.core.fft.fftshift — MLX 0.29.2 documentation + mlx.core.fft.fftshift — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                  mlx.core.fft.fftshift#

                  -fftshift(a: array, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#
                  +fftshift(a: array, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#

                  Shift the zero-frequency component to the center of the spectrum.

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • axes (list(int), optional) – Axes over which to perform the shift. +

                  • axes (list(int), optional) – Axes over which to perform the shift. If None, shift all axes.

                  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 e98cf85df..fb88010a5 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.29.2 documentation + mlx.core.fft.ifft — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

                  mlx.core.fft.ifft#

                  -ifft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array#
                  +ifft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array#

                  One dimensional inverse discrete Fourier Transform.

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • n (int, optional) – Size of the transformed axis. The +

                  • n (int, optional) – Size of the transformed axis. The corresponding axis in the input is truncated or padded with zeros to match n. The default value is a.shape[axis].

                    • -

                  • axis (int, optional) – Axis along which to perform the FFT. The +

                  • axis (int, optional) – Axis along which to perform the FFT. The default is -1.

                  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 38defe529..5743b366a 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.29.2 documentation + mlx.core.fft.ifft2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                  mlx.core.fft.ifft2#

                  -ifft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array#
                  +ifft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array#

                  Two dimensional inverse discrete Fourier Transform.

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • s (list(int), optional) – Sizes of the transformed axes. The +

                  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

                    • -

                  • axes (list(int), optional) – Axes along which to perform the FFT. +

                  • axes (list(int), optional) – Axes along which to perform the FFT. The default is [-2, -1].

                  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 74b3eff8c..38ce3c6fe 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.29.2 documentation + mlx.core.fft.ifftn — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                  mlx.core.fft.ifftn#

                  -ifftn(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#
                  +ifftn(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#

                  n-dimensional inverse discrete Fourier Transform.

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • s (list(int), optional) – Sizes of the transformed axes. The +

                  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

                    • -

                  • axes (list(int), optional) – Axes along which to perform the FFT. +

                  • axes (list(int), optional) – Axes along which to perform the FFT. The default is None in which case the FFT is over the last len(s) axes or all axes if s is also None.

                  diff --git a/docs/build/html/python/_autosummary/mlx.core.fft.ifftshift.html b/docs/build/html/python/_autosummary/mlx.core.fft.ifftshift.html index 949cc078e..7d5c3fbcd 100644 --- a/docs/build/html/python/_autosummary/mlx.core.fft.ifftshift.html +++ b/docs/build/html/python/_autosummary/mlx.core.fft.ifftshift.html @@ -8,7 +8,7 @@ - mlx.core.fft.ifftshift — MLX 0.29.2 documentation + mlx.core.fft.ifftshift — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                  mlx.core.fft.ifftshift#

                  -ifftshift(a: array, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#
                  +ifftshift(a: array, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#

                  The inverse of fftshift(). While identical to fftshift() for even-length axes, the behavior differs for odd-length axes.

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • axes (list(int), optional) – Axes over which to perform the inverse shift. +

                  • axes (list(int), optional) – Axes over which to perform the inverse shift. If None, shift all axes.

                  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 8a29a8202..e98f09936 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.29.2 documentation + mlx.core.fft.irfft — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.fft.irfft#

                  -irfft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array#
                  +irfft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array#

                  The inverse of rfft().

                  The output has the same shape as the input except along axis in which case it has size n.

                  @@ -940,11 +940,11 @@ which case it has size Parameters:

                  • a (array) – The input array.

                    • -

                  • n (int, optional) – Size of the transformed axis. The +

                  • n (int, optional) – Size of the transformed axis. The corresponding axis in the input is truncated or padded with zeros to match n // 2 + 1. The default value is a.shape[axis] // 2 + 1.

                    • -

                  • axis (int, optional) – Axis along which to perform the FFT. The +

                  • axis (int, optional) – Axis along which to perform the FFT. The default is -1.

                  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 8903ef7ae..75f981fff 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.29.2 documentation + mlx.core.fft.irfft2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.fft.irfft2#

                  -irfft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array#
                  +irfft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array#

                  The inverse of rfft2().

                  Note the input is generally complex. The dimensions of the input specified in axes are padded or truncated to match the sizes @@ -942,12 +942,12 @@ and will have size

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • s (list(int), optional) – Sizes of the transformed axes. The +

                  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s except for the last axis which has size s[-1] // 2 + 1. The default value is the sizes of a along axes.

                    • -

                  • axes (list(int), optional) – Axes along which to perform the FFT. +

                  • axes (list(int), optional) – Axes along which to perform the FFT. The default is [-2, -1].

                  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 1c93c7c73..a563d7fec 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.29.2 documentation + mlx.core.fft.irfftn — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.fft.irfftn#

                  -irfftn(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#
                  +irfftn(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#

                  The inverse of rfftn().

                  Note the input is generally complex. The dimensions of the input specified in axes are padded or truncated to match the sizes @@ -942,11 +942,11 @@ and will have size

                  Parameters:

                  • a (array) – The input array.

                    • -

                  • s (list(int), optional) – Sizes of the transformed axes. The +

                  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

                    • -

                  • axes (list(int), optional) – Axes along which to perform the FFT. +

                  • axes (list(int), optional) – Axes along which to perform the FFT. The default is None in which case the FFT is over the last len(s) axes or all axes if s is also None.

                  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 1baa78ab9..d08474e33 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.29.2 documentation + mlx.core.fft.rfft — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.fft.rfft#

                  -rfft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array#
                  +rfft(a: array, n: Optional[int] = None, axis: int = -1, stream: Optional[Union[Stream, Device]] = None) → array#

                  One dimensional discrete Fourier Transform on a real input.

                  The output has the same shape as the input except along axis in which case it has size n // 2 + 1.

                  @@ -941,10 +941,10 @@ which case it has size

                  • a (array) – The input array. If the array is complex it will be silently cast to a real type.

                    • -

                  • n (int, optional) – Size of the transformed axis. The +

                  • n (int, optional) – Size of the transformed axis. The corresponding axis in the input is truncated or padded with zeros to match n. The default value is a.shape[axis].

                    • -

                  • axis (int, optional) – Axis along which to perform the FFT. The +

                  • axis (int, optional) – Axis along which to perform the FFT. The default is -1.

                  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 7fa730b0b..2cb5ee3e9 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.29.2 documentation + mlx.core.fft.rfft2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.fft.rfft2#

                  -rfft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array#
                  +rfft2(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = [-2, -1], stream: Optional[Union[Stream, Device]] = None) → array#

                  Two dimensional real discrete Fourier Transform.

                  The output has the same shape as the input except along the dimensions in axes in which case it has sizes from s. The last axis in axes is @@ -942,11 +942,11 @@ treated as the real axis and will have size

                  • a (array) – The input array. If the array is complex it will be silently cast to a real type.

                    • -

                  • s (list(int), optional) – Sizes of the transformed axes. The +

                  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

                    • -

                  • axes (list(int), optional) – Axes along which to perform the FFT. +

                  • axes (list(int), optional) – Axes along which to perform the FFT. The default is [-2, -1].

                  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 9ec513a7d..90017a926 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.29.2 documentation + mlx.core.fft.rfftn — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.fft.rfftn#

                  -rfftn(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#
                  +rfftn(a: array, s: Optional[tuple[int, ...]] = None, axes: Optional[Sequence[int]] = None, stream: Optional[Union[Stream, Device]] = None) → array#

                  n-dimensional real discrete Fourier Transform.

                  The output has the same shape as the input except along the dimensions in axes in which case it has sizes from s. The last axis in axes is @@ -942,11 +942,11 @@ treated as the real axis and will have size

                  • a (array) – The input array. If the array is complex it will be silently cast to a real type.

                    • -

                  • s (list(int), optional) – Sizes of the transformed axes. The +

                  • s (list(int), optional) – Sizes of the transformed axes. The corresponding axes in the input are truncated or padded with zeros to match the sizes in s. The default value is the sizes of a along axes.

                    • -

                  • axes (list(int), optional) – Axes along which to perform the FFT. +

                  • axes (list(int), optional) – Axes along which to perform the FFT. The default is None in which case the FFT is over the last len(s) axes or all axes if s is also None.

                  diff --git a/docs/build/html/python/_autosummary/mlx.core.finfo.html b/docs/build/html/python/_autosummary/mlx.core.finfo.html index 716081028..56c644ee0 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.29.2 documentation + mlx.core.finfo — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -939,7 +939,7 @@ document.write(`

                  Get information on floating-point types.

                  -__init__(self, arg: Dtype, /) → None#
                  +__init__(self, arg: Dtype, /) → None#

                  Methods

                  diff --git a/docs/build/html/python/_autosummary/mlx.core.flatten.html b/docs/build/html/python/_autosummary/mlx.core.flatten.html index d496710b9..c08150c89 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.29.2 documentation + mlx.core.flatten — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.flatten#

                  -flatten(a: array, /, start_axis: int = 0, end_axis: int = -1, *, stream: None | Stream | Device = None) → array#
                  +flatten(a: array, /, start_axis: int = 0, end_axis: int = -1, *, stream: None | Stream | Device = None) → array#

                  Flatten an array.

                  The axes flattened will be between start_axis and end_axis, inclusive. Negative axes are supported. After converting negative axis to @@ -942,8 +942,8 @@ positive, axes outside the valid range will be clamped to a valid value,

                  Parameters:

                  • a (array) – Input array.

                    • -

                  • start_axis (int, optional) – The first dimension to flatten. Defaults to 0.

                    • -

                  • end_axis (int, optional) – The last dimension to flatten. Defaults to -1.

                    • +

                  • start_axis (int, optional) – The first dimension to flatten. Defaults to 0.

                    • +

                  • end_axis (int, optional) – The last dimension to flatten. Defaults to -1.

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

                  diff --git a/docs/build/html/python/_autosummary/mlx.core.floor.html b/docs/build/html/python/_autosummary/mlx.core.floor.html index 6d2425a32..b5bc321c5 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.29.2 documentation + mlx.core.floor — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.floor#

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

                  Element-wise floor.

                  Parameters:
                  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 0722c74f9..fb3a3fded 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.29.2 documentation + mlx.core.floor_divide — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.floor_divide#

                  -floor_divide(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
                  +floor_divide(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

                  Element-wise integer division.

                  If either array is a floating point type then it is equivalent to calling floor() after divide().

                  diff --git a/docs/build/html/python/_autosummary/mlx.core.full.html b/docs/build/html/python/_autosummary/mlx.core.full.html index 2db1f5e51..08d2b286f 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.29.2 documentation + mlx.core.full — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

                  mlx.core.full#

                  -full(shape: int | Sequence[int], vals: scalar | array, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array#
                  +full(shape: int | Sequence[int], vals: scalar | array, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array#

                  Construct an array with the given value.

                  Constructs an array of size shape filled with vals. If vals is an array it must be broadcastable to the given shape.

                  Parameters:
                    -

                  • shape (int or list(int)) – The shape of the output array.

                    • -

                  • vals (float or int or array) – Values to fill the array with.

                    • +

                  • shape (int or list(int)) – The shape of the output array.

                    • +

                  • vals (float or int or array) – Values to fill the array with.

                  • dtype (Dtype, optional) – Data type of the output array. If unspecified the output type is inferred from vals.

                  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 5177af6ea..cbe4ab918 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.29.2 documentation + mlx.core.gather_mm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.gather_mm#

                  -gather_mm(a: array, b: array, /, lhs_indices: array, rhs_indices: array, *, sorted_indices: bool = False, stream: None | Stream | Device = None) → array#
                  +gather_mm(a: array, b: array, /, lhs_indices: array, rhs_indices: array, *, sorted_indices: bool = False, stream: None | Stream | Device = None) → array#

                  Matrix multiplication with matrix-level gather.

                  Performs a gather of the operands with the given indices followed by a (possibly batched) matrix multiplication of two arrays. This operation @@ -954,7 +954,7 @@ flag can be passed for a possible faster implementation.

                • b (array) – Input array.

                • lhs_indices (array, optional) – Integer indices for a. Default: None

                • rhs_indices (array, optional) – Integer indices for b. Default: None

                  • -

                • sorted_indices (bool, optional) – May allow a faster implementation +

                • sorted_indices (bool, optional) – May allow a faster implementation if the passed indices are sorted. Default: False.

                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 988a99bec..d874da9c9 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.29.2 documentation + mlx.core.gather_qmm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                mlx.core.gather_qmm#

                -gather_qmm(x: array, w: array, /, scales: array, biases: array | None = None, lhs_indices: array | None = None, rhs_indices: array | None = None, transpose: bool = True, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, sorted_indices: bool = False, stream: None | Stream | Device = None) → array#
                +gather_qmm(x: array, w: array, /, scales: array, biases: array | None = None, lhs_indices: array | None = None, rhs_indices: array | None = None, transpose: bool = True, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, sorted_indices: bool = False, stream: None | Stream | Device = None) → array#

                Perform quantized matrix multiplication with matrix-level gather.

                This operation is the quantized equivalent to gather_mm(). Similar to gather_mm(), the indices lhs_indices and @@ -950,15 +950,15 @@ as w elements of w. Default: None.

              • lhs_indices (array, optional) – Integer indices for x. Default: None.

              • rhs_indices (array, optional) – Integer indices for w. Default: None.

                • -

              • transpose (bool, optional) – Defines whether to multiply with the +

              • transpose (bool, optional) – Defines whether to multiply with the transposed w or not, namely whether we are performing x @ w.T or x @ w. Default: True.

                • -

              • group_size (int, optional) – The size of the group in w that +

              • group_size (int, optional) – The size of the group in w that shares a scale and bias. Default: 64.

                • -

              • bits (int, optional) – The number of bits occupied by each element in +

              • bits (int, optional) – The number of bits occupied by each element in w. Default: 4.

                • -

              • mode (str, optional) – The quantization mode. Default: "affine".

                • -

              • sorted_indices (bool, optional) – May allow a faster implementation +

              • mode (str, optional) – The quantization mode. Default: "affine".

                • +

              • sorted_indices (bool, optional) – May allow a faster implementation if the passed indices are sorted. Default: False.

              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 ace245a87..21b0079c0 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.29.2 documentation + mlx.core.get_active_memory — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.get_active_memory#

              -get_active_memory() → int#
              +get_active_memory() → int#

              Get the actively used memory in bytes.

              Note, this will not always match memory use reported by the system because it does not include cached memory buffers.

              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 4315bf809..c52fa7c90 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.29.2 documentation + mlx.core.get_cache_memory — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.get_cache_memory#

              -get_cache_memory() → int#
              +get_cache_memory() → int#

              Get the cache size in bytes.

              The cache includes memory not currently used that has not been returned to the system allocator.

              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 641313459..b5a2ce622 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.29.2 documentation + mlx.core.get_peak_memory — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.get_peak_memory#

              -get_peak_memory() → int#
              +get_peak_memory() → int#

              Get the peak amount of used memory in bytes.

              The maximum memory used recorded from the beginning of the program execution or since the last call to reset_peak_memory().

              diff --git a/docs/build/html/python/_autosummary/mlx.core.grad.html b/docs/build/html/python/_autosummary/mlx.core.grad.html index 84740f43a..27ba1b5ac 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.29.2 documentation + mlx.core.grad — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

              mlx.core.grad#

              -grad(fun: Callable, argnums: int | Sequence[int] | None = None, argnames: str | Sequence[str] = []) → Callable#
              +grad(fun: Callable, argnums: int | Sequence[int] | None = None, argnames: str | Sequence[str] = []) → Callable#

              Returns a function which computes the gradient of fun.

              Parameters:
              @@ -940,12 +940,12 @@ document.write(`

            • fun (Callable) – A function which takes a variable number of array or trees of array and returns a scalar output array.

              • -

            • argnums (int or list(int), optional) – Specify the index (or indices) +

            • argnums (int or list(int), optional) – Specify the index (or indices) of the positional arguments of fun to compute the gradient with respect to. If neither argnums nor argnames are provided argnums defaults to 0 indicating fun’s first argument.

              • -

            • argnames (str or list(str), optional) – Specify keyword arguments of +

            • argnames (str or list(str), optional) – Specify keyword arguments of fun to compute gradients with respect to. It defaults to [] so no gradients for keyword arguments by default.

            diff --git a/docs/build/html/python/_autosummary/mlx.core.greater.html b/docs/build/html/python/_autosummary/mlx.core.greater.html index 8a6ac2020..293bb7d2e 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.29.2 documentation + mlx.core.greater — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.greater#

            -greater(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
            +greater(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

            Element-wise greater than.

            Strict greater than on two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

            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 ee3fceac4..75c28b61b 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.29.2 documentation + mlx.core.greater_equal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.greater_equal#

            -greater_equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
            +greater_equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

            Element-wise greater or equal.

            Greater than or equal on two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

            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 a563a8378..b4c821455 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.29.2 documentation + mlx.core.hadamard_transform — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.hadamard_transform#

            -hadamard_transform(a: array, scale: float | None = None, stream: None | Stream | Device = None) → array#
            +hadamard_transform(a: array, scale: float | None = None, stream: None | Stream | Device = None) → array#

            Perform the Walsh-Hadamard transform along the final axis.

            Equivalent to:

            from scipy.linalg import hadamard
@@ -946,7 +946,7 @@ document.write(`
 
            Parameters:
            
 
              
 
            • a (array) – Input array or scalar.
              • 
-
            • scale (float) – Scale the output by this factor.
+
            • scale (float) – Scale the output by this factor.
 Defaults to 1/sqrt(a.shape[-1]) so that the Hadamard matrix is orthonormal.
              • 
 
            
 
            
diff --git a/docs/build/html/python/_autosummary/mlx.core.identity.html b/docs/build/html/python/_autosummary/mlx.core.identity.html
index ae0c9cb5c..31f8b3d4e 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.29.2 documentation
+    mlx.core.identity — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
            @@ -932,12 +932,12 @@ document.write(`

            mlx.core.identity#

            -identity(n: int, dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#
            +identity(n: int, dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#

            Create a square identity matrix.

            Parameters:
              -

            • n (int) – The number of rows and columns in the output.

              • +

            • n (int) – The number of rows and columns in the output.

            • dtype (Dtype, optional) – Data type of the output array. Defaults to float32.

            • stream (Stream, optional) – Stream or device. Defaults to None.

            diff --git a/docs/build/html/python/_autosummary/mlx.core.imag.html b/docs/build/html/python/_autosummary/mlx.core.imag.html index ea12037a5..75a035766 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.29.2 documentation + mlx.core.imag — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
            @@ -932,7 +932,7 @@ document.write(`

            mlx.core.imag#

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

            Returns the imaginary part of a complex array.

            Parameters:
            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 837604be3..c72399374 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.29.2 documentation + mlx.core.import_function — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.import_function#

            -import_function(file: str) → Callable#
            +import_function(file: str) → Callable#

            Import a function from a file.

            The imported function can be called either with *args and **kwargs or with a tuple of arrays and/or dictionary of string @@ -946,7 +946,7 @@ versions of MLX may not be compatible with future versions.

            Parameters:
            -

            file (str) – The file path to import the function from.

            +

            file (str) – The file path to import the function from.

            Returns:

            The imported function.

            diff --git a/docs/build/html/python/_autosummary/mlx.core.inner.html b/docs/build/html/python/_autosummary/mlx.core.inner.html index 849f0bc61..c3a3cd024 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.29.2 documentation + mlx.core.inner — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.inner#

            -inner(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#
            +inner(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#

            Ordinary inner product of vectors for 1-D arrays, in higher dimensions a sum product over the last axes.

            Parameters:
            diff --git a/docs/build/html/python/_autosummary/mlx.core.isclose.html b/docs/build/html/python/_autosummary/mlx.core.isclose.html index f2b530dfc..2cc0e8052 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.29.2 documentation + mlx.core.isclose — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.isclose#

            -isclose(a: array, b: array, /, rtol: float = 1e-05, atol: float = 1e-08, *, equal_nan: bool = False, stream: None | Stream | Device = None) → array#
            +isclose(a: array, b: array, /, rtol: float = 1e-05, atol: float = 1e-08, *, equal_nan: bool = False, stream: None | Stream | Device = None) → array#

            Returns a boolean array where two arrays are element-wise equal within a tolerance.

            Infinite values are considered equal if they have the same sign, NaN values are not equal unless equal_nan is True.

            @@ -947,9 +947,9 @@ broadcasting.

            • a (array) – Input array.

            • b (array) – Input array.

              • -

            • rtol (float) – Relative tolerance.

              • -

            • atol (float) – Absolute tolerance.

              • -

            • equal_nan (bool) – If True, NaNs are considered equal. +

            • rtol (float) – Relative tolerance.

              • +

            • atol (float) – Absolute tolerance.

              • +

            • equal_nan (bool) – If True, NaNs are considered equal. Defaults to False.

            diff --git a/docs/build/html/python/_autosummary/mlx.core.isfinite.html b/docs/build/html/python/_autosummary/mlx.core.isfinite.html index e63e2c533..670396471 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.29.2 documentation + mlx.core.isfinite — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.isfinite#

            -isfinite(a: array, stream: None | Stream | Device = None) → array#
            +isfinite(a: array, stream: None | Stream | Device = None) → array#

            Return a boolean array indicating which elements are finite.

            An element is finite if it is not infinite or NaN.

            diff --git a/docs/build/html/python/_autosummary/mlx.core.isinf.html b/docs/build/html/python/_autosummary/mlx.core.isinf.html index 9dbc16f0c..93e32616d 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.29.2 documentation + mlx.core.isinf — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.isinf#

            -isinf(a: array, stream: None | Stream | Device = None) → array#
            +isinf(a: array, stream: None | Stream | Device = None) → array#

            Return a boolean array indicating which elements are +/- inifnity.

            Parameters:
            diff --git a/docs/build/html/python/_autosummary/mlx.core.isnan.html b/docs/build/html/python/_autosummary/mlx.core.isnan.html index 9353113f4..7d181a833 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.29.2 documentation + mlx.core.isnan — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.isnan#

            -isnan(a: array, stream: None | Stream | Device = None) → array#
            +isnan(a: array, stream: None | Stream | Device = None) → array#

            Return a boolean array indicating which elements are NaN.

            Parameters:
            diff --git a/docs/build/html/python/_autosummary/mlx.core.isneginf.html b/docs/build/html/python/_autosummary/mlx.core.isneginf.html index da9025184..abb25eafb 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.29.2 documentation + mlx.core.isneginf — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.isneginf#

            -isneginf(a: array, stream: None | Stream | Device = None) → array#
            +isneginf(a: array, stream: None | Stream | Device = None) → array#

            Return a boolean array indicating which elements are negative infinity.

            Parameters:
            diff --git a/docs/build/html/python/_autosummary/mlx.core.isposinf.html b/docs/build/html/python/_autosummary/mlx.core.isposinf.html index 115e1e582..cff948f44 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.29.2 documentation + mlx.core.isposinf — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.isposinf#

            -isposinf(a: array, stream: None | Stream | Device = None) → array#
            +isposinf(a: array, stream: None | Stream | Device = None) → array#

            Return a boolean array indicating which elements are positive infinity.

            Parameters:
            diff --git a/docs/build/html/python/_autosummary/mlx.core.issubdtype.html b/docs/build/html/python/_autosummary/mlx.core.issubdtype.html index 27b348e49..17e7005c9 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.29.2 documentation + mlx.core.issubdtype — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.issubdtype#

            -issubdtype(arg1: Dtype | DtypeCategory, arg2: Dtype | DtypeCategory) → bool#
            +issubdtype(arg1: Dtype | DtypeCategory, arg2: Dtype | DtypeCategory) → bool#

            Check if a Dtype or DtypeCategory is a subtype of another.

            @@ -949,7 +949,7 @@ of another.

            second input.

            Return type:
            -

            bool

            +

            bool

            Example

            diff --git a/docs/build/html/python/_autosummary/mlx.core.jvp.html b/docs/build/html/python/_autosummary/mlx.core.jvp.html index d93f830ce..8aa82976c 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.29.2 documentation + mlx.core.jvp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.jvp#

            -jvp(fun: Callable, primals: list[array], tangents: list[array]) → tuple[list[array], list[array]]#
            +jvp(fun: Callable, primals: list[array], tangents: list[array]) → tuple[list[array], list[array]]#

            Compute the Jacobian-vector product.

            This computes the product of the Jacobian of a function fun evaluated at primals with the tangents.

            @@ -941,9 +941,9 @@ at primals

            • fun (Callable) – A function which takes a variable number of array and returns a single array or list of array.

              • -

            • primals (list(array)) – A list of array at which to +

            • primals (list(array)) – A list of array at which to evaluate the Jacobian.

              • -

            • tangents (list(array)) – A list of array which are the +

            • tangents (list(array)) – A list of array which are the “vector” in the Jacobian-vector product. The tangents should be the same in number, shape, and type as the inputs of fun (i.e. the primals).

            @@ -953,7 +953,7 @@ same in number, shape, and type as the inputs of fun.

            Return type:
            -

            list(array)

            +

            list(array)

            diff --git a/docs/build/html/python/_autosummary/mlx.core.kron.html b/docs/build/html/python/_autosummary/mlx.core.kron.html index 93873557c..c6a005fd3 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.29.2 documentation + mlx.core.kron — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.kron#

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

            Compute the Kronecker product of two arrays a and b.

            Parameters:
            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 84bbba492..a09e9e13b 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.29.2 documentation + mlx.core.left_shift — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.left_shift#

            -left_shift(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
            +left_shift(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

            Element-wise left shift.

            Shift the bits of the first input to the left by the second using numpy-style broadcasting semantics. Either or both input arrays can diff --git a/docs/build/html/python/_autosummary/mlx.core.less.html b/docs/build/html/python/_autosummary/mlx.core.less.html index 4481dd051..f1504fa97 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.29.2 documentation + mlx.core.less — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.less#

            -less(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
            +less(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

            Element-wise less than.

            Strict less than on two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

            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 e85949843..5e3c738db 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.29.2 documentation + mlx.core.less_equal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.less_equal#

            -less_equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
            +less_equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

            Element-wise less than or equal.

            Less than or equal on two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

            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 3ee013dc0..df3942c28 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.29.2 documentation + mlx.core.linalg.cholesky — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

            mlx.core.linalg.cholesky#

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

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

            This function supports arrays with at least 2 dimensions. When the input has more than two dimensions, the Cholesky decomposition is computed for each matrix @@ -942,7 +942,7 @@ in the last two dimensions of Parameters:

            • a (array) – Input array.

              • -

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

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

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

              • 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 79d27ed31..c66e9c38c 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.29.2 documentation + mlx.core.linalg.cholesky_inv — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

              mlx.core.linalg.cholesky_inv#

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

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

              Let \(\mathbf{A}\) be a real symmetric positive semi-definite matrix and \(\mathbf{L}\) its Cholesky decomposition such that:

              @@ -949,7 +949,7 @@ in the last two dimensions of \(\math
              Parameters:

              • L (array) – Input array.

                • -

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

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

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

                • 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 ef3fadb19..86bc153fe 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.29.2 documentation + mlx.core.linalg.cross — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                mlx.core.linalg.cross#

                -cross(a: array, b: array, axis: int = -1, *, stream: None | Stream | Device = None) → array#
                +cross(a: array, b: array, axis: int = -1, *, stream: None | Stream | Device = None) → array#

                Compute the cross product of two arrays along a specified axis.

                The cross product is defined for arrays with size 2 or 3 in the specified axis. If the size is 2 then the third value is assumed @@ -942,7 +942,7 @@ to be zero.

                • a (array) – Input array.

                • b (array) – Input array.

                  • -

                • axis (int, optional) – Axis along which to compute the cross +

                • axis (int, optional) – Axis along which to compute the cross product. Default: -1.

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

                  • diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.eig.html b/docs/build/html/python/_autosummary/mlx.core.linalg.eig.html index e4b11bcea..2ffd38d9f 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.eig.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.eig.html @@ -8,7 +8,7 @@ - mlx.core.linalg.eig — MLX 0.29.2 documentation + mlx.core.linalg.eig — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.linalg.eig#

                  -eig(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array]#
                  +eig(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array]#

                  Compute the eigenvalues and eigenvectors of a square matrix.

                  This function differs from numpy.linalg.eig() in that the return type is always complex even if the eigenvalues are all real.

                  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 31874bb1e..a7580ae21 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.29.2 documentation + mlx.core.linalg.eigh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.linalg.eigh#

                  -eigh(a: array, UPLO: str = 'L', *, stream: None | Stream | Device = None) → Tuple[array, array]#
                  +eigh(a: array, UPLO: str = 'L', *, stream: None | Stream | Device = None) → Tuple[array, array]#

                  Compute the eigenvalues and eigenvectors of a complex Hermitian or real symmetric matrix.

                  This function supports arrays with at least 2 dimensions. When the input @@ -943,7 +943,7 @@ computed for each matrix in the last two dimensions.

                  • a (array) – Input array. Must be a real symmetric or complex Hermitian matrix.

                    • -

                  • UPLO (str, optional) – Whether to use the upper ("U") or +

                  • UPLO (str, optional) – Whether to use the upper ("U") or lower ("L") triangle of the matrix. Default: "L".

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

                    • diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.eigvals.html b/docs/build/html/python/_autosummary/mlx.core.linalg.eigvals.html index f99462e7c..2f1b5acff 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.eigvals.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.eigvals.html @@ -8,7 +8,7 @@ - mlx.core.linalg.eigvals — MLX 0.29.2 documentation + mlx.core.linalg.eigvals — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.linalg.eigvals#

                    -eigvals(a: array, *, stream: Optional[Union[Stream, Device]] = None) → array#
                    +eigvals(a: array, *, stream: Optional[Union[Stream, Device]] = None) → array#

                    Compute the eigenvalues of a square matrix.

                    This function differs from numpy.linalg.eigvals() in that the return type is always complex even if the eigenvalues are all real.

                    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 dc6d200f3..2eee81579 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.29.2 documentation + mlx.core.linalg.eigvalsh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.linalg.eigvalsh#

                    -eigvalsh(a: array, UPLO: str = 'L', *, stream: Optional[Union[Stream, Device]] = None) → array#
                    +eigvalsh(a: array, UPLO: str = 'L', *, stream: Optional[Union[Stream, Device]] = None) → array#

                    Compute the eigenvalues of a complex Hermitian or real symmetric matrix.

                    This function supports arrays with at least 2 dimensions. When the input has more than two dimensions, the eigenvalues are computed for @@ -942,7 +942,7 @@ each matrix in the last two dimensions.

                    • a (array) – Input array. Must be a real symmetric or complex Hermitian matrix.

                      • -

                    • UPLO (str, optional) – Whether to use the upper ("U") or +

                    • UPLO (str, optional) – Whether to use the upper ("U") or lower ("L") triangle of the matrix. Default: "L".

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

                      • 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 a171f6236..5f15b8068 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.29.2 documentation + mlx.core.linalg.inv — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.linalg.inv#

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

                      Compute the inverse of a square matrix.

                      This function supports arrays with at least 2 dimensions. When the input has more than two dimensions, the inverse is computed for each matrix 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 d450df564..77584b72a 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.29.2 documentation + mlx.core.linalg.lu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.linalg.lu#

                      -lu(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array, array]#
                      +lu(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array, array]#

                      Compute the LU factorization of the given matrix A.

                      Note, unlike the default behavior of scipy.linalg.lu, the pivots are indices. To reconstruct the input use L[P, :] @ U for 2 @@ -954,7 +954,7 @@ in which case the default stream of the default device is used.

                      The p, L, and U arrays, such that A = L[P, :] @ U

                      Return type:
                      -

                      tuple(array, array, array)

                      +

                      tuple(array, array, array)

                      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 e96bee587..95372ce3e 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.29.2 documentation + mlx.core.linalg.lu_factor — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.linalg.lu_factor#

                      -lu_factor(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array]#
                      +lu_factor(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array]#

                      Computes a compact representation of the LU factorization.

                      Parameters:
                      @@ -946,7 +946,7 @@ in which case the default stream of the default device is used.

                      The LU matrix and pivots array.

                      Return type:
                      -

                      tuple(array, array)

                      +

                      tuple(array, array)

                      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 b750d3353..1d69a9730 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.29.2 documentation + mlx.core.linalg.norm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                      mlx.core.linalg.norm#

                      -norm(a: array, /, ord: None | int | float | str = None, axis: None | int | list[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
                      +norm(a: array, /, ord: None | int | float | str = None, axis: None | int | list[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

                      Matrix or vector norm.

                      This function computes vector or matrix norms depending on the value of the ord and axis parameters.

                      @@ -943,16 +943,16 @@ the ord

                      a (array) – Input array. If axis is None, a must be 1-D or 2-D, unless ord is None. If both axis and ord are None, the 2-norm of a.flatten will be returned.

                      -

                    • ord (int, float or str, optional) – Order of the norm (see table under Notes). +

                    • ord (int, float or str, optional) – Order of the norm (see table under Notes). If None, the 2-norm (or Frobenius norm for matrices) will be computed along the given axis. Default: None.

                      • -

                    • axis (int or list(int), optional) – If axis is an integer, it specifies the +

                    • axis (int or list(int), optional) – If axis is an integer, it specifies the axis of a along which to compute the vector norms. If axis is a 2-tuple, it specifies the axes that hold 2-D matrices, and the matrix norms of these matrices are computed. If axis is None then either a vector norm (when a is 1-D) or a matrix norm (when a is 2-D) is returned. Default: None.

                      • -

                    • keepdims (bool, optional) – If True, the axes which are normed over are +

                    • keepdims (bool, optional) – If True, the axes which are normed over are left in the result as dimensions with size one. Default False.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.pinv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.pinv.html index f245331c3..cf9735f0e 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.pinv.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.pinv.html @@ -8,7 +8,7 @@ - mlx.core.linalg.pinv — MLX 0.29.2 documentation + mlx.core.linalg.pinv — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.linalg.pinv#

                    -pinv(a: array, *, stream: None | Stream | Device = None) → array#
                    +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. 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 d111a9c84..a954dae28 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.29.2 documentation + mlx.core.linalg.qr — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.linalg.qr#

                    -qr(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array]#
                    +qr(a: array, *, stream: None | Stream | Device = None) → Tuple[array, array]#

                    The QR factorization of the input matrix.

                    This function supports arrays with at least 2 dimensions. The matrices which are factorized are assumed to be in the last two dimensions of @@ -949,7 +949,7 @@ in which case the default stream of the default device is used.

                    Q and R matrices such that Q @ R = a.

                    Return type:
                    -

                    tuple(array, array)

                    +

                    tuple(array, array)

                    Example

                    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 c23a87d95..45dabdf90 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.29.2 documentation + mlx.core.linalg.solve — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.linalg.solve#

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

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

                    Parameters:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.solve_triangular.html b/docs/build/html/python/_autosummary/mlx.core.linalg.solve_triangular.html index 30e103ae4..9b48f06b1 100644 --- a/docs/build/html/python/_autosummary/mlx.core.linalg.solve_triangular.html +++ b/docs/build/html/python/_autosummary/mlx.core.linalg.solve_triangular.html @@ -8,7 +8,7 @@ - mlx.core.linalg.solve_triangular — MLX 0.29.2 documentation + mlx.core.linalg.solve_triangular — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                    mlx.core.linalg.solve_triangular#

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

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

                    Parameters:

                    • a (array) – Input array.

                    • b (array) – Input array.

                      • -

                    • upper (bool, optional) – Whether the array is upper or lower +

                    • upper (bool, optional) – Whether the array is upper or lower triangular. Default: False.

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

                      • 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 cef24b647..87a8f7849 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.29.2 documentation + mlx.core.linalg.svd — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.linalg.svd#

                      -svd(a: array, compute_uv: bool = True, *, stream: None | Stream | Device = None) → Tuple[array, array, array]#
                      +svd(a: array, compute_uv: bool = True, *, stream: None | Stream | Device = None) → Tuple[array, array, array]#

                      The Singular Value Decomposition (SVD) of the input matrix.

                      This function supports arrays with at least 2 dimensions. When the input has more than two dimensions, the function iterates over all indices of the first @@ -941,7 +941,7 @@ a.ndim - 2 dimensions and for each combination SVD is applied to the last two in

                      Parameters:

                      • a (array) – Input array.

                        • -

                      • compute_uv (bool, optional) – If True, return the U, S, and Vt components. +

                      • compute_uv (bool, optional) – If True, return the U, S, and Vt components. If False, return only the S array. Default: True.

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

                        • @@ -952,7 +952,7 @@ in which case the default stream of the default device is used.

                        A = U @ diag(S) @ Vt. If compute_uv is False returns singular values array S.

                      Return type:
                      -

                      Union[tuple(array, …), array]

                      +

                      Union[tuple(array, …), array]

                    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 fec897f23..72e30fa47 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.29.2 documentation + mlx.core.linalg.tri_inv — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.linalg.tri_inv#

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

                    Compute the inverse of a triangular square matrix.

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

                    • a (array) – Input array.

                      • -

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

                      • +

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

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

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.linspace.html b/docs/build/html/python/_autosummary/mlx.core.linspace.html index f18745dbf..ff746f752 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.29.2 documentation + mlx.core.linspace — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                    mlx.core.linspace#

                    -linspace(start, stop, num: int | None = 50, dtype: Dtype | None = float32, stream: None | Stream | Device = None) → array#
                    +linspace(start, stop, num: int | None = 50, dtype: Dtype | None = float32, stream: None | Stream | Device = None) → array#

                    Generate num evenly spaced numbers over interval [start, stop].

                    Parameters:

                    • start (scalar) – Starting value.

                    • stop (scalar) – Stopping value.

                      • -

                    • num (int, optional) – Number of samples, defaults to 50.

                      • +

                    • num (int, optional) – Number of samples, defaults to 50.

                    • dtype (Dtype, optional) – Specifies the data type of the output, default to float32.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.load.html b/docs/build/html/python/_autosummary/mlx.core.load.html index c2510e32c..9ea6a5d72 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.29.2 documentation + mlx.core.load — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,18 +932,18 @@ document.write(`

                    mlx.core.load#

                    -load(file: file | str | Path, /, format: str | None = None, return_metadata: bool = False, *, stream: None | Stream | Device = None) → array | dict[str, array]#
                    +load(file: file | str | Path, /, format: str | None = None, return_metadata: bool = False, *, stream: None | Stream | Device = None) → array | dict[str, array]#

                    Load array(s) from a binary file.

                    The supported formats are .npy, .npz, .safetensors, and .gguf.

                    Parameters:
                      -

                    • file (file, str, Path) – File in which the array is saved.

                      • -

                    • format (str, optional) – Format of the file. If None, the +

                    • file (file, str, Path) – File in which the array is saved.

                      • +

                    • format (str, optional) – Format of the file. If None, the format is inferred from the file extension. Supported formats: npy, npz, and safetensors. Default: None.

                      • -

                    • return_metadata (bool, optional) – Load the metadata for formats +

                    • return_metadata (bool, optional) – Load the metadata for formats which support matadata. The metadata will be returned as an additional dictionary. Default: False.

                    @@ -955,7 +955,7 @@ mapping names to arrays if loading from a Return type: -

                    array or dict

                    +

                    array or dict

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.log.html b/docs/build/html/python/_autosummary/mlx.core.log.html index 3236c853a..18be5e2da 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.29.2 documentation + mlx.core.log — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                    @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.log#

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

                    Element-wise natural logarithm.

                    Parameters:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.log10.html b/docs/build/html/python/_autosummary/mlx.core.log10.html index e1496f750..c752e6b58 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.29.2 documentation + mlx.core.log10 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.log10#

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

                    Element-wise base-10 logarithm.

                    Parameters:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.log1p.html b/docs/build/html/python/_autosummary/mlx.core.log1p.html index ba958c591..54da27693 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.29.2 documentation + mlx.core.log1p — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.log1p#

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

                    Element-wise natural log of one plus the array.

                    Parameters:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.log2.html b/docs/build/html/python/_autosummary/mlx.core.log2.html index d76fe036e..e6e389c07 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.29.2 documentation + mlx.core.log2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.log2#

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

                    Element-wise base-2 logarithm.

                    Parameters:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.logaddexp.html b/docs/build/html/python/_autosummary/mlx.core.logaddexp.html index ba28248a3..8d2c68b67 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.29.2 documentation + mlx.core.logaddexp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.logaddexp#

                    -logaddexp(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) → array#
                    +logaddexp(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) → array#

                    Element-wise log-add-exp.

                    This is a numerically stable log-add-exp of two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.logcumsumexp.html b/docs/build/html/python/_autosummary/mlx.core.logcumsumexp.html index 1cc696eb5..20ab9af93 100644 --- a/docs/build/html/python/_autosummary/mlx.core.logcumsumexp.html +++ b/docs/build/html/python/_autosummary/mlx.core.logcumsumexp.html @@ -8,7 +8,7 @@ - mlx.core.logcumsumexp — MLX 0.29.2 documentation + mlx.core.logcumsumexp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                    mlx.core.logcumsumexp#

                    -logcumsumexp(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#
                    +logcumsumexp(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array#

                    Return the cumulative logsumexp of the elements along the given axis.

                    Parameters:

                    • a (array) – Input array

                      • -

                    • axis (int, optional) – Optional axis to compute the cumulative logsumexp +

                    • axis (int, optional) – Optional axis to compute the cumulative logsumexp over. If unspecified the cumulative logsumexp of the flattened array is returned.

                      • -

                    • reverse (bool) – Perform the cumulative logsumexp in reverse.

                      • -

                    • inclusive (bool) – The i-th element of the output includes the i-th +

                    • reverse (bool) – Perform the cumulative logsumexp in reverse.

                      • +

                    • inclusive (bool) – The i-th element of the output includes the i-th element of the input.

                    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 fd4e3b211..450d4e632 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.29.2 documentation + mlx.core.logical_and — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.logical_and#

                    -logical_and(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#
                    +logical_and(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#

                    Element-wise logical and.

                    Parameters:
                    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 0f2246a47..a700fdd9c 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.29.2 documentation + mlx.core.logical_not — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.logical_not#

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

                    Element-wise logical not.

                    Parameters:
                    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 8ba800e97..873c32909 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.29.2 documentation + mlx.core.logical_or — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.logical_or#

                    -logical_or(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#
                    +logical_or(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#

                    Element-wise logical or.

                    Parameters:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.logsumexp.html b/docs/build/html/python/_autosummary/mlx.core.logsumexp.html index c6037452e..f1683e975 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.29.2 documentation + mlx.core.logsumexp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.logsumexp#

                    -logsumexp(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
                    +logsumexp(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

                    A log-sum-exp reduction over the given axes.

                    The log-sum-exp reduction is a numerically stable version of:

                    log(sum(exp(a), axis))
@@ -942,10 +942,10 @@ document.write(`
 
                    Parameters:
                    
 
                      
 
                    • a (array) – Input array.
                      • 
-
                    • axis (int or list(int), optional) – Optional axis or
+
                    • axis (int or list(int), optional) – Optional axis or
 axes to reduce over. If unspecified this defaults
 to reducing over the entire array.
                      • 
-
                    • keepdims (bool, optional) – Keep reduced axes as
+
                    • keepdims (bool, optional) – Keep reduced axes as
 singleton dimensions, defaults to False.
                      • 
 
                    
 
                    
diff --git a/docs/build/html/python/_autosummary/mlx.core.matmul.html b/docs/build/html/python/_autosummary/mlx.core.matmul.html
index 3044a0cf2..7c2e0ff94 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.29.2 documentation
+    mlx.core.matmul — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                    @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.matmul#

                    -matmul(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#
                    +matmul(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#

                    Matrix multiplication.

                    Perform the (possibly batched) matrix multiplication of two arrays. This function supports broadcasting for arrays with more than two dimensions.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.max.html b/docs/build/html/python/_autosummary/mlx.core.max.html index 4d961651b..2fbb64192 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.29.2 documentation + mlx.core.max — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                    @@ -932,16 +932,16 @@ document.write(`

                    mlx.core.max#

                    -max(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
                    +max(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

                    A max reduction over the given axes.

                    Parameters:

                    • a (array) – Input array.

                      • -

                    • axis (int or list(int), optional) – Optional axis or +

                    • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

                      • -

                    • keepdims (bool, optional) – Keep reduced axes as +

                    • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.maximum.html b/docs/build/html/python/_autosummary/mlx.core.maximum.html index 4f2ad70ec..42bfab7c4 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.29.2 documentation + mlx.core.maximum — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.maximum#

                    -maximum(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) → array#
                    +maximum(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) → array#

                    Element-wise maximum.

                    Take the element-wise max of two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.mean.html b/docs/build/html/python/_autosummary/mlx.core.mean.html index a5283b0f5..10216761b 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.29.2 documentation + mlx.core.mean — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

                    mlx.core.mean#

                    -mean(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
                    +mean(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

                    Compute the mean(s) over the given axes.

                    Parameters:

                    • a (array) – Input array.

                      • -

                    • axis (int or list(int), optional) – Optional axis or +

                    • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

                      • -

                    • keepdims (bool, optional) – Keep reduced axes as +

                    • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.meshgrid.html b/docs/build/html/python/_autosummary/mlx.core.meshgrid.html index e49e97052..9437c03c3 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.29.2 documentation + mlx.core.meshgrid — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

                    mlx.core.meshgrid#

                    -meshgrid(*arrays: array, sparse: bool | None = False, indexing: str | None = 'xy', stream: None | Stream | Device = None) → array#
                    +meshgrid(*arrays: array, sparse: bool | None = False, indexing: str | None = 'xy', stream: None | Stream | Device = None) → array#

                    Generate multidimensional coordinate grids from 1-D coordinate arrays

                    Parameters:

                    • *arrays (array) – Input arrays.

                      • -

                    • sparse (bool, optional) – If True, a sparse grid is returned in which each output +

                    • sparse (bool, optional) – If True, a sparse grid is returned in which each output array has a single non-zero element. If False, a dense grid is returned. Defaults to False.

                      • -

                    • indexing (str, optional) – Cartesian (‘xy’) or matrix (‘ij’) indexing of the output arrays. +

                    • indexing (str, optional) – Cartesian (‘xy’) or matrix (‘ij’) indexing of the output arrays. Defaults to 'xy'.

                    @@ -949,7 +949,7 @@ Defaults to 'xy'

                    The output arrays.

                    Return type:
                    -

                    list(array)

                    +

                    list(array)

                    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 65bf7a32b..c785f67d4 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.29.2 documentation + mlx.core.metal.device_info — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.metal.device_info#

                    -device_info() → dict[str, Union[str, int]]#
                    +device_info() → dict[str, Union[str, int]]#

                    Get information about the GPU device and system settings.

                    Currently returns:

                      @@ -947,7 +947,7 @@ document.write(`

                      A dictionary with string keys and string or integer values.

                      Return type:
                      -

                      dict

                      +

                      dict

                    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 4f978234b..c3a9a5136 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.29.2 documentation + mlx.core.metal.is_available — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.metal.is_available#

                    -is_available() → bool#
                    +is_available() → bool#

                    Check if the Metal back-end is available.

                    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 7dc085587..ad4e7f604 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.29.2 documentation + mlx.core.metal.start_capture — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,11 +932,11 @@ document.write(`

                    mlx.core.metal.start_capture#

                    -start_capture(path: str) → None#
                    +start_capture(path: str) → None#

                    Start a Metal capture.

                    Parameters:
                    -

                    path (str) – The path to save the capture which should have +

                    path (str) – The path to save the capture which should have the extension .gputrace.

                    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 38bae1049..b431333ca 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.29.2 documentation + mlx.core.metal.stop_capture — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.metal.stop_capture#

                    -stop_capture() → None#
                    +stop_capture() → None#

                    Stop a Metal capture.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.min.html b/docs/build/html/python/_autosummary/mlx.core.min.html index 163c9794d..1594d8b33 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.29.2 documentation + mlx.core.min — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

                    mlx.core.min#

                    -min(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
                    +min(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

                    A min reduction over the given axes.

                    Parameters:

                    • a (array) – Input array.

                      • -

                    • axis (int or list(int), optional) – Optional axis or +

                    • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

                      • -

                    • keepdims (bool, optional) – Keep reduced axes as +

                    • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.minimum.html b/docs/build/html/python/_autosummary/mlx.core.minimum.html index 6decb87a4..a9269d047 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.29.2 documentation + mlx.core.minimum — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.minimum#

                    -minimum(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) → array#
                    +minimum(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) → array#

                    Element-wise minimum.

                    Take the element-wise min of two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.moveaxis.html b/docs/build/html/python/_autosummary/mlx.core.moveaxis.html index cd6346906..de48bd65b 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.29.2 documentation + mlx.core.moveaxis — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                    mlx.core.moveaxis#

                    -moveaxis(a: array, /, source: int, destination: int, *, stream: None | Stream | Device = None) → array#
                    +moveaxis(a: array, /, source: int, destination: int, *, stream: None | Stream | Device = None) → array#

                    Move an axis to a new position.

                    Parameters:

                    • a (array) – Input array.

                      • -

                    • source (int) – Specifies the source axis.

                      • -

                    • destination (int) – Specifies the destination axis.

                      • +

                    • source (int) – Specifies the source axis.

                      • +

                    • destination (int) – Specifies the destination axis.

                    Returns:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.multiply.html b/docs/build/html/python/_autosummary/mlx.core.multiply.html index 65e9ef79a..99f21941e 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.29.2 documentation + mlx.core.multiply — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.multiply#

                    -multiply(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
                    +multiply(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

                    Element-wise multiplication.

                    Multiply two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

                    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 f996420c6..40a027d26 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.29.2 documentation + mlx.core.nan_to_num — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                    mlx.core.nan_to_num#

                    -nan_to_num(a: scalar | array, nan: float = 0, posinf: float | None = None, neginf: float | None = None, *, stream: None | Stream | Device = None) → array#
                    +nan_to_num(a: scalar | array, nan: float = 0, posinf: float | None = None, neginf: float | None = None, *, stream: None | Stream | Device = None) → array#

                    Replace NaN and Inf values with finite numbers.

                    Parameters:

                    • a (array) – Input array

                      • -

                    • nan (float, optional) – Value to replace NaN with. Default: 0.

                      • -

                    • posinf (float, optional) – Value to replace positive infinities +

                    • nan (float, optional) – Value to replace NaN with. Default: 0.

                      • +

                    • posinf (float, optional) – Value to replace positive infinities with. If None, defaults to largest finite value for the given data type. Default: None.

                      • -

                    • neginf (float, optional) – Value to replace negative infinities +

                    • neginf (float, optional) – Value to replace negative infinities with. If None, defaults to the negative of the largest finite value for the given data type. Default: None.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.negative.html b/docs/build/html/python/_autosummary/mlx.core.negative.html index f4d4ae78a..1cbd9e60c 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.29.2 documentation + mlx.core.negative — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.negative#

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

                    Element-wise negation.

                    Parameters:
                    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 7d0f65143..3506afe94 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.29.2 documentation + mlx.core.new_stream — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 b33de99dc..ff86889f0 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.29.2 documentation + mlx.core.not_equal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.not_equal#

                    -not_equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
                    +not_equal(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

                    Element-wise not equal.

                    Not equal comparison on two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.ones.html b/docs/build/html/python/_autosummary/mlx.core.ones.html index 7a32df6f1..a22f8f386 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.29.2 documentation + mlx.core.ones — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,12 +932,12 @@ document.write(`

                    mlx.core.ones#

                    -ones(shape: int | Sequence[int], dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#
                    +ones(shape: int | Sequence[int], dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#

                    Construct an array of ones.

                    Parameters:
                      -

                    • shape (int or list(int)) – The shape of the output array.

                      • +

                    • shape (int or list(int)) – The shape of the output array.

                    • dtype (Dtype, optional) – Data type of the output array. If unspecified the output type defaults to float32.

                    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 6420b5919..7afa940b1 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.29.2 documentation + mlx.core.ones_like — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.ones_like#

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

                    An array of ones like the input.

                    Parameters:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.outer.html b/docs/build/html/python/_autosummary/mlx.core.outer.html index e047d2b35..28595a4aa 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.29.2 documentation + mlx.core.outer — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.outer#

                    -outer(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#
                    +outer(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#

                    Compute the outer product of two 1-D arrays, if the array’s passed are not 1-D a flatten op will be run beforehand.

                    Parameters:
                    diff --git a/docs/build/html/python/_autosummary/mlx.core.pad.html b/docs/build/html/python/_autosummary/mlx.core.pad.html index 2abe0985a..4adab8b0c 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.29.2 documentation + mlx.core.pad — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                    mlx.core.pad#

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

                    Pad an array with a constant value

                    Parameters:

                    • a (array) – Input array.

                      • -

                    • pad_width (int, tuple(int), tuple(int, int) or list(tuple(int, int))) – Number of padded +

                    • pad_width (int, tuple(int), tuple(int, int) or list(tuple(int, int))) – Number of padded values to add to the edges of each axis:((before_1, after_1), (before_2, after_2), ..., (before_N, after_N)). If a single pair of integers is passed then (before_i, after_i) are all the same. diff --git a/docs/build/html/python/_autosummary/mlx.core.partition.html b/docs/build/html/python/_autosummary/mlx.core.partition.html index f8d1c4177..aa4f0eb3a 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.29.2 documentation + mlx.core.partition — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.partition#

                      -partition(a: array, /, kth: int, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#
                      +partition(a: array, /, kth: int, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#

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

                      The ordering of the elements in partitions is undefined.

                      @@ -940,11 +940,11 @@ elements are first.

                      Parameters:

                      • a (array) – Input array.

                        • -

                      • kth (int) – Element at the kth index will be in its sorted +

                      • kth (int) – Element at the kth index will be in its sorted position in the output. All elements before the kth index will be less or equal to the kth element and all elements after will be greater or equal to the kth element in the output.

                        • -

                      • axis (int or None, optional) – Optional axis to partition over. +

                      • axis (int or None, optional) – Optional axis to partition over. If None, this partitions over the flattened array. If unspecified, it defaults to -1.

                      diff --git a/docs/build/html/python/_autosummary/mlx.core.power.html b/docs/build/html/python/_autosummary/mlx.core.power.html index 0eae5c9c5..bdb8a3514 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.29.2 documentation + mlx.core.power — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.power#

                      -power(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) → array#
                      +power(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) → array#

                      Element-wise power operation.

                      Raise the elements of a to the powers in elements of b with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

                      diff --git a/docs/build/html/python/_autosummary/mlx.core.prod.html b/docs/build/html/python/_autosummary/mlx.core.prod.html index e192557e3..90e75904d 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.29.2 documentation + mlx.core.prod — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

                      mlx.core.prod#

                      -prod(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
                      +prod(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

                      An product reduction over the given axes.

                      Parameters:

                      • a (array) – Input array.

                        • -

                      • axis (int or list(int), optional) – Optional axis or +

                      • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

                        • -

                      • keepdims (bool, optional) – Keep reduced axes as +

                      • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

                      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 6347aae9c..68ca85325 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.29.2 documentation + mlx.core.put_along_axis — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.put_along_axis#

                      -put_along_axis(a: array, /, indices: array, values: array, axis: int | None = None, *, stream: None | Stream | Device = None) → array#
                      +put_along_axis(a: array, /, indices: array, values: array, axis: int | None = None, *, stream: None | Stream | Device = None) → array#

                      Put values along an axis at the specified indices.

                      Parameters:
                      @@ -942,7 +942,7 @@ document.write(` the input array excluding the axis dimension.

                    • values (array) – Values array. These should be broadcastable with the indices.

                      • -

                    • axis (int or None) – Axis in the destination to put the values to. If +

                    • axis (int or None) – Axis in the destination to put the values to. If axis == None the destination is flattened prior to the put operation.

                    diff --git a/docs/build/html/python/_autosummary/mlx.core.quantize.html b/docs/build/html/python/_autosummary/mlx.core.quantize.html index dedfc36a5..bcd1bfbdf 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.29.2 documentation + mlx.core.quantize — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                    mlx.core.quantize#

                    -quantize(w: array, /, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, stream: None | Stream | Device = None) → tuple[array, array, array]#
                    +quantize(w: array, /, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, stream: None | Stream | Device = None) → tuple[array, array, array]#

                    Quantize the matrix w using bits bits per element.

                    Note, every group_size elements in a row of w are quantized together. Hence, number of columns of w should be divisible by @@ -950,11 +950,11 @@ are described in more detail below.

                    Parameters:

                    • w (array) – Matrix to be quantized

                      • -

                    • group_size (int, optional) – The size of the group in w that shares a +

                    • group_size (int, optional) – The size of the group in w that shares a scale and bias. Default: 64.

                      • -

                    • bits (int, optional) – The number of bits occupied by each element of +

                    • bits (int, optional) – The number of bits occupied by each element of w in the returned quantized matrix. Default: 4.

                      • -

                    • mode (str, optional) – The quantization mode. Default: "affine".

                      • +

                    • mode (str, optional) – The quantization mode. Default: "affine".

                    Returns:
                    @@ -967,7 +967,7 @@ scale and bias. Default:
                    Return type:
                    -

                    tuple

                    +

                    tuple

                    Notes

                    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 f00880ea4..856d0aad8 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.29.2 documentation + mlx.core.quantized_matmul — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                    mlx.core.quantized_matmul#

                    -quantized_matmul(x: array, w: array, /, scales: array, biases: array | None = None, transpose: bool = True, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, stream: None | Stream | Device = None) → array#
                    +quantized_matmul(x: array, w: array, /, scales: array, biases: array | None = None, transpose: bool = True, group_size: int = 64, bits: int = 4, mode: str = 'affine', *, stream: None | Stream | Device = None) → array#

                    Perform the matrix multiplication with the quantized matrix w. The quantization uses one floating point scale and bias per group_size of elements. Each element in w takes bits bits and is packed in an @@ -945,14 +945,14 @@ unsigned 32 bit integer.

                  • scales (array) – The scales to use per group_size elements of w

                  • biases (array, optional) – The biases to use per group_size elements of w. Default: None.

                    • -

                  • transpose (bool, optional) – Defines whether to multiply with the +

                  • transpose (bool, optional) – Defines whether to multiply with the transposed w or not, namely whether we are performing x @ w.T or x @ w. Default: True.

                    • -

                  • group_size (int, optional) – The size of the group in w that +

                  • group_size (int, optional) – The size of the group in w that shares a scale and bias. Default: 64.

                    • -

                  • bits (int, optional) – The number of bits occupied by each element in +

                  • bits (int, optional) – The number of bits occupied by each element in w. Default: 4.

                    • -

                  • mode (str, optional) – The quantization mode. Default: "affine".

                    • +

                  • mode (str, optional) – The quantization mode. Default: "affine".

                  Returns:
                  diff --git a/docs/build/html/python/_autosummary/mlx.core.radians.html b/docs/build/html/python/_autosummary/mlx.core.radians.html index 1a7d5ab09..6ab2887a8 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.29.2 documentation + mlx.core.radians — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.radians#

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

                  Convert angles from degrees to radians.

                  Parameters:
                  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 175e39299..e14facbd1 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.29.2 documentation + mlx.core.random.bernoulli — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                  mlx.core.random.bernoulli#

                  -bernoulli(p: scalar | array = 0.5, shape: Sequence[int] | None = None, key: array | None = None, stream: None | Stream | Device = None) → array#
                  +bernoulli(p: scalar | array = 0.5, shape: Sequence[int] | None = None, key: array | None = None, stream: None | Stream | Device = None) → array#

                  Generate Bernoulli random values.

                  The values are sampled from the bernoulli distribution with parameter -p. The parameter p can be a float or array and +p. The parameter p can be a float or array and must be broadcastable to shape.

                  Parameters:
                    -

                  • p (float or array, optional) – Parameter of the Bernoulli +

                  • p (float or array, optional) – Parameter of the Bernoulli distribution. Default: 0.5.

                    • -

                  • shape (list(int), optional) – Shape of the output. +

                  • shape (list(int), optional) – Shape of the output. Default: p.shape.

                  • key (array, optional) – A PRNG key. Default: None.

                  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 ca25b76e6..01e07329f 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.29.2 documentation + mlx.core.random.categorical — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                  mlx.core.random.categorical#

                  -categorical(logits: array, axis: int = -1, shape: Sequence[int] | None = None, num_samples: int | None = None, key: array | None = None, stream: None | Stream | Device = None) → array#
                  +categorical(logits: array, axis: int = -1, shape: Sequence[int] | None = None, num_samples: int | None = None, key: array | None = None, stream: None | Stream | Device = None) → array#

                  Sample from a categorical distribution.

                  The values are sampled from the categorical distribution specified by the unnormalized values in logits. Note, at most one of shape @@ -942,12 +942,12 @@ has the same shape as Parameters:

                  • logits (array) – The unnormalized categorical distribution(s).

                    • -

                  • axis (int, optional) – The axis which specifies the distribution. +

                  • axis (int, optional) – The axis which specifies the distribution. Default: -1.

                    • -

                  • shape (list(int), optional) – The shape of the output. This must +

                  • shape (list(int), optional) – The shape of the output. This must be broadcast compatible with logits.shape with the axis dimension removed. Default: None

                    • -

                  • num_samples (int, optional) – The number of samples to draw from each +

                  • num_samples (int, optional) – The number of samples to draw from each of the categorical distributions in logits. The output will have num_samples in the last dimension. Default: None.

                  • key (array, optional) – A PRNG key. Default: None.

                    • 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 b04e894ed..6d7acec20 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.29.2 documentation + mlx.core.random.gumbel — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                    mlx.core.random.gumbel#

                    -gumbel(shape: Sequence[int] = [], dtype: Dtype | None = float32, key: None | Stream | Device = None, stream: array | None = None) → array#
                    +gumbel(shape: Sequence[int] = [], dtype: Dtype | None = float32, key: None | Stream | Device = None, stream: array | None = None) → array#

                    Sample from the standard Gumbel distribution.

                    The values are sampled from a standard Gumbel distribution which CDF exp(-exp(-x)).

                    Parameters:
                      -

                    • shape (list(int)) – The shape of the output.

                      • +

                    • shape (list(int)) – The shape of the output.

                    • dtype (Dtype, optional) – The data type of the output. Default: float32.

                    • key (array, optional) – A PRNG key. Default: None.

                      • 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 5263606b9..75e4b7e6c 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.29.2 documentation + mlx.core.random.key — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,11 +932,11 @@ document.write(`

                      mlx.core.random.key#

                      -key(seed: int) → array#
                      +key(seed: int) → array#

                      Get a PRNG key from a seed.

                      Parameters:
                      -

                      seed (int) – Seed for the PRNG.

                      +

                      seed (int) – Seed for the PRNG.

                      Returns:

                      The PRNG key array.

                      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 70ee5534c..f20d5aa92 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.29.2 documentation + mlx.core.random.laplace — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

                      mlx.core.random.laplace#

                      -laplace(shape: Sequence[int] = [], dtype: Dtype | None = float32, loc: float = 0.0, scale: float = 1.0, key: array | None = None, stream: None | Stream | Device = None) → array#
                      +laplace(shape: Sequence[int] = [], dtype: Dtype | None = float32, loc: float = 0.0, scale: float = 1.0, key: array | None = None, stream: None | Stream | Device = None) → array#

                      Sample numbers from a Laplace distribution.

                      Parameters:
                        -

                      • shape (list(int), optional) – Shape of the output. Default: ().

                        • +

                      • shape (list(int), optional) – Shape of the output. Default: ().

                      • dtype (Dtype, optional) – Type of the output. Default: float32.

                        • -

                      • loc (float, optional) – Mean of the distribution. Default: 0.0.

                        • -

                      • scale (float, optional) – The scale “b” of the Laplace distribution. +

                      • loc (float, optional) – Mean of the distribution. Default: 0.0.

                        • +

                      • scale (float, optional) – The scale “b” of the Laplace distribution. Default:1.0.

                      • key (array, optional) – A PRNG key. Default: None.

                      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 ef2ff099a..8e4fff23d 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.29.2 documentation + mlx.core.random.multivariate_normal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.random.multivariate_normal#

                      -multivariate_normal(mean: array, cov: array, shape: Sequence[int] = [], dtype: Dtype | None = float32, key: array | None = None, stream: None | Stream | Device = None) → array#
                      +multivariate_normal(mean: array, cov: array, shape: Sequence[int] = [], dtype: Dtype | None = float32, key: array | None = None, stream: None | Stream | Device = None) → array#

                      Generate jointly-normal random samples given a mean and covariance.

                      The matrix cov must be positive semi-definite. The behavior is undefined if it is not. The only supported dtype is float32.

                      @@ -944,7 +944,7 @@ distribution.

                    • cov (array) – array of shape (..., n, n), the covariance matrix of the distribution. The batch shape ... must be broadcast-compatible with that of mean.

                      • -

                    • shape (list(int), optional) – The output shape must be +

                    • shape (list(int), optional) – The output shape must be broadcast-compatible with mean.shape[:-1] and cov.shape[:-2]. If empty, the result shape is determined by broadcasting the batch shapes of mean and cov. Default: [].

                      • 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 bfa17ff18..eef960728 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.29.2 documentation + mlx.core.random.normal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.random.normal#

                      -normal(shape: Sequence[int] = [], dtype: Dtype | None = float32, loc: scalar | array | None = None, scale: scalar | array | None = None, key: array | None = None, stream: None | Stream | Device = None) → array#
                      +normal(shape: Sequence[int] = [], dtype: Dtype | None = float32, loc: scalar | array | None = None, scale: scalar | array | None = None, key: array | None = None, stream: None | Stream | Device = None) → array#

                      Generate normally distributed random numbers.

                      If loc and scale are not provided the “standard” normal distribution is used. That means $x sim mathcal{N}(0, 1)$ for @@ -941,7 +941,7 @@ frac{1}{2})$ for complex numbers.

                      Parameters:
                        -

                      • shape (list(int), optional) – Shape of the output. Default: ().

                        • +

                      • shape (list(int), optional) – Shape of the output. Default: ().

                      • dtype (Dtype, optional) – Type of the output. Default: float32.

                      • loc (scalar or array, optional) – Mean of the distribution. Default: None.

                        • 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 c6b12fd65..a7f275174 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.29.2 documentation + mlx.core.random.permutation — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

                        mlx.core.random.permutation#

                        -permutation(x: int | array, axis: int = 0, key: array | None = None, stream: None | Stream | Device = None) → array#
                        +permutation(x: int | array, axis: int = 0, key: array | None = None, stream: None | Stream | Device = None) → array#

                        Generate a random permutation or permute the entries of an array.

                        Parameters:
                          -

                        • x (int or array, optional) – If an integer is provided a random +

                        • x (int or array, optional) – If an integer is provided a random permtuation of mx.arange(x) is returned. Otherwise the entries of x along the given axis are randomly permuted.

                          • -

                        • axis (int, optional) – The axis to permute along. Default: 0.

                          • +

                        • axis (int, optional) – The axis to permute along. Default: 0.

                        • key (array, optional) – A PRNG key. Default: None.

                        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 ee1950cc8..f2d743e4c 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.29.2 documentation + mlx.core.random.randint — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.random.randint#

                        -randint(low: scalar | array, high: scalar | array, shape: Sequence[int] = [], dtype: Dtype | None = int32, key: array | None = None, stream: None | Stream | Device = None) → array#
                        +randint(low: scalar | array, high: scalar | array, shape: Sequence[int] = [], dtype: Dtype | None = int32, key: array | None = None, stream: None | Stream | Device = None) → array#

                        Generate random integers from the given interval.

                        The values are sampled with equal probability from the integers in half-open interval [low, high). The lower and upper bound can be @@ -942,7 +942,7 @@ scalars or arrays and must be broadcastable to

                        • low (scalar or array) – Lower bound of the interval.

                        • high (scalar or array) – Upper bound of the interval.

                          • -

                        • shape (list(int), optional) – Shape of the output. Default: ().

                          • +

                        • shape (list(int), optional) – Shape of the output. Default: ().

                        • dtype (Dtype, optional) – Type of the output. Default: int32.

                        • key (array, optional) – A PRNG key. Default: None.

                        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 4aca72853..ac17bd2c1 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.29.2 documentation + mlx.core.random.seed — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,11 +932,11 @@ document.write(`

                        mlx.core.random.seed#

                        -seed(seed: int) → None#
                        +seed(seed: int) → None#

                        Seed the global PRNG.

                        Parameters:
                        -

                        seed (int) – Seed for the global PRNG.

                        +

                        seed (int) – Seed for the global PRNG.

                        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 eade6f8ed..131d94d9e 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.29.2 documentation + mlx.core.random.split — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                        mlx.core.random.split#

                        -split(key: array, num: int = 2, stream: None | Stream | Device = None) → array#
                        +split(key: array, num: int = 2, stream: None | Stream | Device = None) → array#

                        Split a PRNG key into sub keys.

                        Parameters:

                        • key (array) – Input key to split.

                          • -

                        • num (int, optional) – Number of sub keys. Default: 2.

                          • +

                        • num (int, optional) – Number of sub keys. Default: 2.

                        Returns:
                        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 ad8ffc3f4..6ec2820f8 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.29.2 documentation + mlx.core.random.truncated_normal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.random.truncated_normal#

                        -truncated_normal(lower: scalar | array, upper: scalar | array, shape: Sequence[int] | None = None, dtype: Dtype | None = float32, key: array | None = None, stream: None | Stream | Device = None) → array#
                        +truncated_normal(lower: scalar | array, upper: scalar | array, shape: Sequence[int] | None = None, dtype: Dtype | None = float32, key: array | None = None, stream: None | Stream | Device = None) → array#

                        Generate values from a truncated normal distribution.

                        The values are sampled from the truncated normal distribution on the domain (lower, upper). The bounds lower and upper @@ -942,7 +942,7 @@ can be scalars or arrays and must be broadcastable to

                        • lower (scalar or array) – Lower bound of the domain.

                        • upper (scalar or array) – Upper bound of the domain.

                          • -

                        • shape (list(int), optional) – The shape of the output. +

                        • shape (list(int), optional) – The shape of the output. Default:().

                        • dtype (Dtype, optional) – The data type of the output. Default: float32.

                          • 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 60180364a..199a62569 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.29.2 documentation + mlx.core.random.uniform — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.random.uniform#

                          -uniform(low: scalar | array = 0, high: scalar | array = 1, shape: Sequence[int] = [], dtype: Dtype | None = float32, key: array | None = None, stream: None | Stream | Device = None) → array#
                          +uniform(low: scalar | array = 0, high: scalar | array = 1, shape: Sequence[int] = [], dtype: Dtype | None = float32, key: array | None = None, stream: None | Stream | Device = None) → array#

                          Generate uniformly distributed random numbers.

                          The values are sampled uniformly in the half-open interval [low, high). The lower and upper bound can be scalars or arrays and must be @@ -944,7 +944,7 @@ broadcastable to sh Default: 0.

                        • high (scalar or array, optional) – Upper bound of the distribution. Default: 1.

                          • -

                        • shape (list(int), optional) – Shape of the output. Default:().

                          • +

                        • shape (list(int), optional) – Shape of the output. Default:().

                        • dtype (Dtype, optional) – Type of the output. Default: float32.

                        • key (array, optional) – A PRNG key. Default: None.

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.real.html b/docs/build/html/python/_autosummary/mlx.core.real.html index 79bcfe0fa..b1b2e7bde 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.29.2 documentation + mlx.core.real — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.real#

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

                        Returns the real part of a complex array.

                        Parameters:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.reciprocal.html b/docs/build/html/python/_autosummary/mlx.core.reciprocal.html index 421a91337..5c166f596 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.29.2 documentation + mlx.core.reciprocal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.reciprocal#

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

                        Element-wise reciprocal.

                        Parameters:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.remainder.html b/docs/build/html/python/_autosummary/mlx.core.remainder.html index 5bbe63230..31723797e 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.29.2 documentation + mlx.core.remainder — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.remainder#

                        -remainder(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
                        +remainder(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

                        Element-wise remainder of division.

                        Computes the remainder of dividing a with b with numpy-style broadcasting semantics. Either or both input arrays can also be diff --git a/docs/build/html/python/_autosummary/mlx.core.repeat.html b/docs/build/html/python/_autosummary/mlx.core.repeat.html index ae3b9cde8..aa352d3aa 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.29.2 documentation + mlx.core.repeat — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                        mlx.core.repeat#

                        -repeat(array: array, repeats: int, axis: int | None = None, *, stream: None | Stream | Device = None) → array#
                        +repeat(array: array, repeats: int, axis: int | None = None, *, stream: None | Stream | Device = None) → array#

                        Repeat an array along a specified axis.

                        Parameters:

                        • array (array) – Input array.

                          • -

                        • repeats (int) – The number of repetitions for each element.

                          • -

                        • axis (int, optional) – The axis in which to repeat the array along. If +

                        • repeats (int) – The number of repetitions for each element.

                          • +

                        • axis (int, optional) – The axis in which to repeat the array along. If unspecified it uses the flattened array of the input and repeats along axis 0.

                        • stream (Stream, optional) – Stream or device. Defaults to None.

                          • 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 ff742a2a4..f722a594a 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.29.2 documentation + mlx.core.reset_peak_memory — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.reset_peak_memory#

                          -reset_peak_memory() → None#
                          +reset_peak_memory() → None#

                          Reset the peak memory to zero.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.reshape.html b/docs/build/html/python/_autosummary/mlx.core.reshape.html index 22e189f3e..24f4fb8f2 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.29.2 documentation + mlx.core.reshape — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                          mlx.core.reshape#

                          -reshape(a: array, /, shape: Sequence[int], *, stream: None | Stream | Device = None) → array#
                          +reshape(a: array, /, shape: Sequence[int], *, stream: None | Stream | Device = None) → array#

                          Reshape an array while preserving the size.

                          Parameters:

                          • a (array) – Input array.

                            • -

                          • shape (tuple(int)) – New shape.

                            • +

                          • shape (tuple(int)) – New shape.

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

                          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 6e81ad780..b7b1ee48f 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.29.2 documentation + mlx.core.right_shift — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.right_shift#

                          -right_shift(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
                          +right_shift(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

                          Element-wise right shift.

                          Shift the bits of the first input to the right by the second using numpy-style broadcasting semantics. Either or both input arrays can diff --git a/docs/build/html/python/_autosummary/mlx.core.roll.html b/docs/build/html/python/_autosummary/mlx.core.roll.html index dcd11e193..34d100a30 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.29.2 documentation + mlx.core.roll — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.roll#

                          -roll(a: array, shift: int | Tuple[int], axis: None | int | Tuple[int] = None, /, *, stream: None | Stream | Device = None) → array#
                          +roll(a: array, shift: int | Tuple[int], axis: None | int | Tuple[int] = None, /, *, stream: None | Stream | Device = None) → array#

                          Roll array elements along a given axis.

                          Elements that are rolled beyond the end of the array are introduced at the beggining and vice-versa.

                          @@ -942,11 +942,11 @@ shape is restored.

                          Parameters:

                          • a (array) – Input array

                            • -

                          • shift (int or tuple(int)) – The number of places by which elements +

                          • shift (int or tuple(int)) – The number of places by which elements are shifted. If positive the array is rolled to the right, if negative it is rolled to the left. If an int is provided but the axis is a tuple then the same value is used for all axes.

                            • -

                          • axis (int or tuple(int), optional) – The axis or axes along which to +

                          • axis (int or tuple(int), optional) – The axis or axes along which to roll the elements.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.round.html b/docs/build/html/python/_autosummary/mlx.core.round.html index 2143911ad..da49fcf61 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.29.2 documentation + mlx.core.round — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.round#

                          -round(a: array, /, decimals: int = 0, stream: None | Stream | Device = None) → array#
                          +round(a: array, /, decimals: int = 0, stream: None | Stream | Device = None) → array#

                          Round to the given number of decimals.

                          Basically performs:

                          s = 10**decimals
@@ -943,7 +943,7 @@ document.write(`
 
                          Parameters:
                          
 
                            
 
                          • a (array) – Input array
                            • 
-
                          • decimals (int) – Number of decimal places to round to. (default: 0)
                            • 
+
                          • decimals (int) – Number of decimal places to round to. (default: 0)
                            • 
 
                          
 
                          
 
                          Returns:
                          
diff --git a/docs/build/html/python/_autosummary/mlx.core.rsqrt.html b/docs/build/html/python/_autosummary/mlx.core.rsqrt.html
index c8d819682..987eb835c 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.29.2 documentation
+    mlx.core.rsqrt — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                          @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.rsqrt#

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

                          Element-wise reciprocal and square root.

                          Parameters:
                          diff --git a/docs/build/html/python/_autosummary/mlx.core.save.html b/docs/build/html/python/_autosummary/mlx.core.save.html index e76fd4e82..84b3773b2 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.29.2 documentation + mlx.core.save — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                          @@ -932,12 +932,12 @@ document.write(`

                          mlx.core.save#

                          -save(file: file | str | Path, arr: array) → None#
                          +save(file: file | str | Path, arr: array) → None#

                          Save the array to a binary file in .npy format.

                          Parameters:
                            -

                          • file (str, Path, file) – File to which the array is saved

                            • +

                          • file (str, Path, file) – File to which the array is saved

                          • arr (array) – Array to be saved.

                          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 c884c1722..984e50547 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.29.2 documentation + mlx.core.save_gguf — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,19 +932,19 @@ document.write(`

                          mlx.core.save_gguf#

                          -save_gguf(file: file | str | Path, arrays: dict[str, array], metadata: dict[str, array | str | list[str]])#
                          +save_gguf(file: file | str | Path, arrays: dict[str, array], metadata: dict[str, array | str | list[str]])#

                          Save array(s) to a binary file in .gguf format.

                          See the GGUF documentation for more information on the format.

                          Parameters:
                            -

                          • file (file, str, Path) – File in which the array is saved.

                            • -

                          • arrays (dict(str, array)) – The dictionary of names to arrays to +

                          • file (file, str, Path) – File in which the array is saved.

                            • +

                          • arrays (dict(str, array)) – The dictionary of names to arrays to be saved.

                            • -

                          • metadata (dict(str, Union[array, str, list(str)])) – The dictionary +

                          • metadata (dict(str, Union[array, str, list(str)])) – The dictionary of metadata to be saved. The values can be a scalar or 1D -obj:array, a str, or a list of str.

                            • +obj:array, a str, or a list of str.

                          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 8c1dbc01f..1f43ebc99 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.29.2 documentation + mlx.core.save_safetensors — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                          mlx.core.save_safetensors#

                          -save_safetensors(file: file | str | Path, arrays: dict[str, array], metadata: dict[str, str] | None = None)#
                          +save_safetensors(file: file | str | Path, arrays: dict[str, array], metadata: dict[str, str] | None = None)#

                          Save array(s) to a binary file in .safetensors format.

                          See the Safetensors documentation for more information on the format.

                          Parameters:
                            -

                          • file (file, str, Path) – File in which the array is saved.

                            • -

                          • arrays (dict(str, array)) – The dictionary of names to arrays to +

                          • file (file, str, Path) – File in which the array is saved.

                            • +

                          • arrays (dict(str, array)) – The dictionary of names to arrays to be saved.

                            • -

                          • metadata (dict(str, str), optional) – The dictionary of +

                          • metadata (dict(str, str), optional) – The dictionary of metadata to be saved.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.savez.html b/docs/build/html/python/_autosummary/mlx.core.savez.html index 1ba4606bb..5b138e0d8 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.29.2 documentation + mlx.core.savez — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.savez#

                          -savez(file: file | str | Path, *args, **kwargs)#
                          +savez(file: file | str | Path, *args, **kwargs)#

                          Save several arrays to a binary file in uncompressed .npz format.

                          import mlx.core as mx
@@ -951,7 +951,7 @@ format.
                          
 
                          
 
                          Parameters:
                          
 
                            
-
                          • file (file, str, Path) – Path to file to which the arrays are saved.
                            • 
+
                          • file (file, str, Path) – Path to file to which the arrays are saved.
                            • 
 
                          • *args (arrays) – Arrays to be saved.
                            • 
 
                          • **kwargs (arrays) – Arrays to be saved. Each array will be saved
 with the associated keyword as the output file name.
                            • 
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 166eca651..98cfe399a 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.29.2 documentation
+    mlx.core.savez_compressed — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                          @@ -932,12 +932,12 @@ document.write(`

                          mlx.core.savez_compressed#

                          -savez_compressed(file: file | str | Path, *args, **kwargs)#
                          +savez_compressed(file: file | str | Path, *args, **kwargs)#

                          Save several arrays to a binary file in compressed .npz format.

                          Parameters:
                            -

                          • file (file, str, Path) – Path to file to which the arrays are saved.

                            • +

                          • file (file, str, Path) – Path to file to which the arrays are saved.

                          • *args (arrays) – Arrays to be saved.

                          • **kwargs (arrays) – Arrays to be saved. Each array will be saved with the associated keyword as the output file name.

                            • 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 6f660c4f3..052485a57 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.29.2 documentation + mlx.core.set_cache_limit — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                          @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.set_cache_limit#

                          -set_cache_limit(limit: int) → int#
                          +set_cache_limit(limit: int) → int#

                          Set the free cache limit.

                          If using more than the given limit, free memory will be reclaimed from the cache on the next allocation. To disable the cache, set @@ -941,13 +941,13 @@ the limit to 0set_memory_limit() for more details.

                          Parameters:
                          -

                          limit (int) – The cache limit in bytes.

                          +

                          limit (int) – The cache limit in bytes.

                          Returns:

                          The previous cache limit in bytes.

                          Return type:
                          -

                          int

                          +

                          int

                          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 ecac45385..4f8c69864 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.29.2 documentation + mlx.core.set_default_device — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.set_default_device#

                          -set_default_device(device: Device) → None#
                          +set_default_device(device: Device) → None#

                          Set the default device.

                          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 6561cd772..5444e7011 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.29.2 documentation + mlx.core.set_default_stream — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.set_default_stream#

                          -set_default_stream(stream: Stream) → None#
                          +set_default_stream(stream: Stream) → None#

                          Set the default stream.

                          This will make the given stream the default for the streams device. It will not change the default device.

                          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 722c1bad8..3deacdef2 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.29.2 documentation + mlx.core.set_memory_limit — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.set_memory_limit#

                          -set_memory_limit(limit: int) → int#
                          +set_memory_limit(limit: int) → int#

                          Set the memory limit.

                          The memory limit is a guideline for the maximum amount of memory to use during graph evaluation. If the memory limit is exceeded and there is no @@ -942,13 +942,13 @@ exception.

                          maximum recommended working set size reported by the device.

                          Parameters:
                          -

                          limit (int) – Memory limit in bytes.

                          +

                          limit (int) – Memory limit in bytes.

                          Returns:

                          The previous memory limit in bytes.

                          Return type:
                          -

                          int

                          +

                          int

                          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 2eca1abaa..c642ccba4 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.29.2 documentation + mlx.core.set_wired_limit — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.set_wired_limit#

                          -set_wired_limit(limit: int) → int#
                          +set_wired_limit(limit: int) → int#

                          Set the wired size limit.

                          Note

                          @@ -954,13 +954,13 @@ increase the system wired limit with:

                          ("memory_size").

                          Parameters:
                          -

                          limit (int) – The wired limit in bytes.

                          +

                          limit (int) – The wired limit in bytes.

                          Returns:

                          The previous wired limit in bytes.

                          Return type:
                          -

                          int

                          +

                          int

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.sigmoid.html b/docs/build/html/python/_autosummary/mlx.core.sigmoid.html index 26b7216b7..c4e3943cb 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.29.2 documentation + mlx.core.sigmoid — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                          mlx.core.sigmoid#

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

                          Element-wise logistic sigmoid.

                          The logistic sigmoid function is:

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.sign.html b/docs/build/html/python/_autosummary/mlx.core.sign.html index f252b3745..3c89cbbef 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.29.2 documentation + mlx.core.sign — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                          @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.sign#

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

                          Element-wise sign.

                          Parameters:
                          diff --git a/docs/build/html/python/_autosummary/mlx.core.sin.html b/docs/build/html/python/_autosummary/mlx.core.sin.html index 7e8990ddf..93905857b 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.29.2 documentation + mlx.core.sin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.sin#

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

                          Element-wise sine.

                          Parameters:
                          diff --git a/docs/build/html/python/_autosummary/mlx.core.sinh.html b/docs/build/html/python/_autosummary/mlx.core.sinh.html index 9cf4a63f5..fde40d371 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.29.2 documentation + mlx.core.sinh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.sinh#

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

                          Element-wise hyperbolic sine.

                          Parameters:
                          diff --git a/docs/build/html/python/_autosummary/mlx.core.slice.html b/docs/build/html/python/_autosummary/mlx.core.slice.html index 800a693ac..ec4ad5d79 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.29.2 documentation + mlx.core.slice — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

                          mlx.core.slice#

                          -slice(a: array, start_indices: array, axes: Sequence[int], slice_size: Sequence[int], *, stream: None | Stream | Device = None) → array#
                          +slice(a: array, start_indices: array, axes: Sequence[int], slice_size: Sequence[int], *, stream: None | Stream | Device = None) → array#

                          Extract a sub-array from the input array.

                          Parameters:

                          • a (array) – Input array

                          • start_indices (array) – The index location to start the slice at.

                            • -

                          • axes (tuple(int)) – The axes corresponding to the indices in start_indices.

                            • -

                          • slice_size (tuple(int)) – The size of the slice.

                            • +

                          • axes (tuple(int)) – The axes corresponding to the indices in start_indices.

                            • +

                          • slice_size (tuple(int)) – The size of the slice.

                          Returns:
                          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 c72441367..b321abd5e 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.29.2 documentation + mlx.core.slice_update — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.slice_update#

                          -slice_update(a: array, update: array, start_indices: array, axes: Sequence[int], *, stream: None | Stream | Device = None) → array#
                          +slice_update(a: array, update: array, start_indices: array, axes: Sequence[int], *, stream: None | Stream | Device = None) → array#

                          Update a sub-array of the input array.

                          Parameters:
                          @@ -940,7 +940,7 @@ document.write(`

                        • a (array) – The input array to update

                        • update (array) – The update array.

                        • start_indices (array) – The index location to start the slice at.

                          • -

                        • axes (tuple(int)) – The axes corresponding to the indices in start_indices.

                          • +

                        • axes (tuple(int)) – The axes corresponding to the indices in start_indices.

                        Returns:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.softmax.html b/docs/build/html/python/_autosummary/mlx.core.softmax.html index b75f45951..aec209710 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.29.2 documentation + mlx.core.softmax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.softmax#

                        -softmax(a: array, /, axis: None | int | Sequence[int] = None, *, stream: None | Stream | Device = None) → array#
                        +softmax(a: array, /, axis: None | int | Sequence[int] = None, *, stream: None | Stream | Device = None) → array#

                        Perform the softmax along the given axis.

                        This operation is a numerically stable version of:

                        exp(a) / sum(exp(a), axis, keepdims=True)
@@ -942,7 +942,7 @@ document.write(`
 
                        Parameters:
                        
 
                          
 
                        • a (array) – Input array.
                          • 
-
                        • axis (int or list(int), optional) – Optional axis or axes to compute
+
                        • axis (int or list(int), optional) – Optional axis or axes to compute
 the softmax over. If unspecified this performs the softmax over
 the full array.
                          • 
 
                        
diff --git a/docs/build/html/python/_autosummary/mlx.core.sort.html b/docs/build/html/python/_autosummary/mlx.core.sort.html
index 844aa458a..251f12fd9 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.29.2 documentation
+    mlx.core.sort — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                        @@ -932,13 +932,13 @@ document.write(`

                        mlx.core.sort#

                        -sort(a: array, /, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#
                        +sort(a: array, /, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#

                        Returns a sorted copy of the array.

                        Parameters:

                        • a (array) – Input array.

                          • -

                        • axis (int or None, optional) – Optional axis to sort over. +

                        • axis (int or None, optional) – Optional axis to sort over. If None, this sorts over the flattened array. If unspecified, it defaults to -1 (sorting over the last axis).

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.split.html b/docs/build/html/python/_autosummary/mlx.core.split.html index 6cc65daf3..f810a99b5 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.29.2 documentation + mlx.core.split — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                        @@ -932,25 +932,25 @@ document.write(`

                        mlx.core.split#

                        -split(a: array, /, indices_or_sections: int | Sequence[int], axis: int = 0, *, stream: None | Stream | Device = None) → array#
                        +split(a: array, /, indices_or_sections: int | Sequence[int], axis: int = 0, *, stream: None | Stream | Device = None) → array#

                        Split an array along a given axis.

                        Parameters:

                        • a (array) – Input array.

                          • -

                        • indices_or_sections (int or list(int)) – If indices_or_sections +

                        • indices_or_sections (int or list(int)) – If indices_or_sections is an integer the array is split into that many sections of equal size. An error is raised if this is not possible. If indices_or_sections is a list, the list contains the indices of the start of each subarray along the given axis.

                          • -

                        • axis (int, optional) – Axis to split along, defaults to 0.

                          • +

                        • axis (int, optional) – Axis to split along, defaults to 0.

                        Returns:

                        A list of split arrays.

                        Return type:
                        -

                        list(array)

                        +

                        list(array)

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.sqrt.html b/docs/build/html/python/_autosummary/mlx.core.sqrt.html index ad132dad0..cb78767bf 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.29.2 documentation + mlx.core.sqrt — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.sqrt#

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

                        Element-wise square root.

                        Parameters:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.square.html b/docs/build/html/python/_autosummary/mlx.core.square.html index 8267ba55b..3be0cb188 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.29.2 documentation + mlx.core.square — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.square#

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

                        Element-wise square.

                        Parameters:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.squeeze.html b/docs/build/html/python/_autosummary/mlx.core.squeeze.html index d1af02e31..163bf2c11 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.29.2 documentation + mlx.core.squeeze — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                        mlx.core.squeeze#

                        -squeeze(a: array, /, axis: None | int | Sequence[int] = None, *, stream: None | Stream | Device = None) → array#
                        +squeeze(a: array, /, axis: None | int | Sequence[int] = None, *, stream: None | Stream | Device = None) → array#

                        Remove length one axes from an array.

                        Parameters:

                        • a (array) – Input array.

                          • -

                        • axis (int or tuple(int), optional) – Axes to remove. Defaults +

                        • axis (int or tuple(int), optional) – Axes to remove. Defaults to None in which case all size one axes are removed.

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.stack.html b/docs/build/html/python/_autosummary/mlx.core.stack.html index 84aae33a7..e639b833c 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.29.2 documentation + mlx.core.stack — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                        mlx.core.stack#

                        -stack(arrays: list[array], axis: int | None = 0, *, stream: None | Stream | Device = None) → array#
                        +stack(arrays: list[array], axis: int | None = 0, *, stream: None | Stream | Device = None) → array#

                        Stacks the arrays along a new axis.

                        Parameters:
                          -

                        • arrays (list(array)) – A list of arrays to stack.

                          • -

                        • axis (int, optional) – The axis in the result array along which the +

                        • arrays (list(array)) – A list of arrays to stack.

                          • +

                        • axis (int, optional) – The axis in the result array along which the input arrays are stacked. Defaults to 0.

                        • stream (Stream, optional) – Stream or device. Defaults to None.

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.std.html b/docs/build/html/python/_autosummary/mlx.core.std.html index d3e64b01a..5b76e4366 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.29.2 documentation + mlx.core.std — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,18 +932,18 @@ document.write(`

                        mlx.core.std#

                        -std(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, ddof: int = 0, *, stream: None | Stream | Device = None) → array#
                        +std(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, ddof: int = 0, *, stream: None | Stream | Device = None) → array#

                        Compute the standard deviation(s) over the given axes.

                        Parameters:

                        • a (array) – Input array.

                          • -

                        • axis (int or list(int), optional) – Optional axis or +

                        • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

                          • -

                        • keepdims (bool, optional) – Keep reduced axes as +

                        • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

                          • -

                        • ddof (int, optional) – The divisor to compute the variance +

                        • ddof (int, optional) – The divisor to compute the variance is N - ddof, defaults to 0.

                        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 83434130e..8c066b4e6 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.29.2 documentation + mlx.core.stop_gradient — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.stop_gradient#

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

                        Stop gradients from being computed.

                        The operation is the identity but it prevents gradients from flowing through the array.

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.stream.html b/docs/build/html/python/_autosummary/mlx.core.stream.html index 9c1702490..a4e78e7f7 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.29.2 documentation + mlx.core.stream — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.stream#

                        -stream(s: Union[Stream, Device]) → mlx.core.StreamContext#
                        +stream(s: Union[Stream, Device]) → mlx.core.StreamContext#

                        Create a context manager to set the default device and stream.

                        Parameters:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.subtract.html b/docs/build/html/python/_autosummary/mlx.core.subtract.html index dd1989bdd..ea9165ded 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.29.2 documentation + mlx.core.subtract — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.subtract#

                        -subtract(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#
                        +subtract(a: scalar | array, b: scalar | array, stream: None | Stream | Device = None) → array#

                        Element-wise subtraction.

                        Subtract one array from another with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.sum.html b/docs/build/html/python/_autosummary/mlx.core.sum.html index fbf3f690a..99eb55839 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.29.2 documentation + mlx.core.sum — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

                        mlx.core.sum#

                        -sum(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#
                        +sum(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array#

                        Sum reduce the array over the given axes.

                        Parameters:

                        • a (array) – Input array.

                          • -

                        • axis (int or list(int), optional) – Optional axis or +

                        • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

                          • -

                        • keepdims (bool, optional) – Keep reduced axes as +

                        • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.swapaxes.html b/docs/build/html/python/_autosummary/mlx.core.swapaxes.html index 780a37f13..34bac06b1 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.29.2 documentation + mlx.core.swapaxes — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                        mlx.core.swapaxes#

                        -swapaxes(a: array, /, axis1: int, axis2: int, *, stream: None | Stream | Device = None) → array#
                        +swapaxes(a: array, /, axis1: int, axis2: int, *, stream: None | Stream | Device = None) → array#

                        Swap two axes of an array.

                        Parameters:

                        • a (array) – Input array.

                          • -

                        • axis1 (int) – Specifies the first axis.

                          • -

                        • axis2 (int) – Specifies the second axis.

                          • +

                        • axis1 (int) – Specifies the first axis.

                          • +

                        • axis2 (int) – Specifies the second axis.

                        Returns:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.synchronize.html b/docs/build/html/python/_autosummary/mlx.core.synchronize.html index 45912ada0..63ecc4b3f 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.29.2 documentation + mlx.core.synchronize — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.synchronize#

                        -synchronize(stream: Optional[Stream] = None) → None#
                        +synchronize(stream: Optional[Stream] = None) → None#

                        Synchronize with the given stream.

                        Parameters:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.take.html b/docs/build/html/python/_autosummary/mlx.core.take.html index 6dc55d84a..0bc218625 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.29.2 documentation + mlx.core.take — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.take#

                        -take(a: array, /, indices: int | array, axis: int | None = None, *, stream: None | Stream | Device = None) → array#
                        +take(a: array, /, indices: int | array, axis: int | None = None, *, stream: None | Stream | Device = None) → array#

                        Take elements along an axis.

                        The elements are taken from indices along the specified axis. If the axis is not specified the array is treated as a flattened @@ -942,8 +942,8 @@ If the axis is not specified the array is treated as a flattened

                        Parameters:

                        • a (array) – Input array.

                          • -

                        • indices (int or array) – Integer index or input array with integral type.

                          • -

                        • axis (int, optional) – Axis along which to perform the take. If unspecified +

                        • indices (int or array) – Integer index or input array with integral type.

                          • +

                        • axis (int, optional) – Axis along which to perform the take. If unspecified the array is treated as a flattened 1-D vector.

                        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 9784348f8..01109d071 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.29.2 documentation + mlx.core.take_along_axis — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                        mlx.core.take_along_axis#

                        -take_along_axis(a: array, /, indices: array, axis: int | None = None, *, stream: None | Stream | Device = None) → array#
                        +take_along_axis(a: array, /, indices: array, axis: int | None = None, *, stream: None | Stream | Device = None) → array#

                        Take values along an axis at the specified indices.

                        Parameters:
                        @@ -940,7 +940,7 @@ document.write(`

                      • a (array) – Input array.

                      • indices (array) – Indices array. These should be broadcastable with the input array excluding the axis dimension.

                        • -

                      • axis (int or None) – Axis in the input to take the values from. If +

                      • axis (int or None) – Axis in the input to take the values from. If axis == None the array is flattened to 1D prior to the indexing operation.

                      diff --git a/docs/build/html/python/_autosummary/mlx.core.tan.html b/docs/build/html/python/_autosummary/mlx.core.tan.html index 32686c2fe..cee98a9b3 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.29.2 documentation + mlx.core.tan — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.tan#

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

                      Element-wise tangent.

                      Parameters:
                      diff --git a/docs/build/html/python/_autosummary/mlx.core.tanh.html b/docs/build/html/python/_autosummary/mlx.core.tanh.html index 8dfb359bc..a16185924 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.29.2 documentation + mlx.core.tanh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                      mlx.core.tanh#

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

                      Element-wise hyperbolic tangent.

                      Parameters:
                      diff --git a/docs/build/html/python/_autosummary/mlx.core.tensordot.html b/docs/build/html/python/_autosummary/mlx.core.tensordot.html index f07c55d6e..8ff90536f 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.29.2 documentation + mlx.core.tensordot — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                      mlx.core.tensordot#

                      -tensordot(a: array, b: array, /, axes: int | list[Sequence[int]] = 2, *, stream: None | Stream | Device = None) → array#
                      +tensordot(a: array, b: array, /, axes: int | list[Sequence[int]] = 2, *, stream: None | Stream | Device = None) → array#

                      Compute the tensor dot product along the specified axes.

                      Parameters:

                      • a (array) – Input array

                      • b (array) – Input array

                        • -

                      • axes (int or list(list(int)), optional) – The number of dimensions to +

                      • axes (int or list(list(int)), optional) – The number of dimensions to sum over. If an integer is provided, then sum over the last axes dimensions of a and the first axes dimensions of b. If a list of lists is provided, then sum over the diff --git a/docs/build/html/python/_autosummary/mlx.core.tile.html b/docs/build/html/python/_autosummary/mlx.core.tile.html index 5bd57f9d2..d2d10f37a 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.29.2 documentation + mlx.core.tile — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                        mlx.core.tile#

                        -tile(a: array, reps: int | Sequence[int], /, *, stream: None | Stream | Device = None) → array#
                        +tile(a: array, reps: int | Sequence[int], /, *, stream: None | Stream | Device = None) → array#

                        Construct an array by repeating a the number of times given by reps.

                        Parameters:

                        • a (array) – Input array

                          • -

                        • reps (int or list(int)) – The number of times to repeat a along each axis.

                          • +

                        • reps (int or list(int)) – The number of times to repeat a along each axis.

                        Returns:
                        diff --git a/docs/build/html/python/_autosummary/mlx.core.topk.html b/docs/build/html/python/_autosummary/mlx.core.topk.html index 4a75f1ad0..0033804cc 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.29.2 documentation + mlx.core.topk — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

                        mlx.core.topk#

                        -topk(a: array, /, k: int, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#
                        +topk(a: array, /, k: int, axis: None | int = -1, *, stream: None | Stream | Device = None) → array#

                        Returns the k largest elements from the input along a given axis.

                        The elements will not necessarily be in sorted order.

                        Parameters:

                        • a (array) – Input array.

                          • -

                        • k (int) – k top elements to be returned

                          • -

                        • axis (int or None, optional) – Optional axis to select over. +

                        • k (int) – k top elements to be returned

                          • +

                        • axis (int or None, optional) – Optional axis to select over. If None, this selects the top k elements over the flattened array. If unspecified, it defaults to -1.

                        diff --git a/docs/build/html/python/_autosummary/mlx.core.trace.html b/docs/build/html/python/_autosummary/mlx.core.trace.html index a0479922d..ca25c42e4 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.29.2 documentation + mlx.core.trace — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                        mlx.core.trace#

                        -trace(a: array, /, offset: int = 0, axis1: int = 0, axis2: int = 1, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array#
                        +trace(a: array, /, offset: int = 0, axis1: int = 0, axis2: int = 1, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array#

                        Return the sum along a specified diagonal in the given array.

                        Parameters:

                        • a (array) – Input array

                          • -

                        • offset (int, optional) – Offset of the diagonal from the main diagonal. +

                        • offset (int, optional) – Offset of the diagonal from the main diagonal. Can be positive or negative. Default: 0.

                          • -

                        • axis1 (int, optional) – The first axis of the 2-D sub-arrays from which +

                        • axis1 (int, optional) – The first axis of the 2-D sub-arrays from which the diagonals should be taken. Default: 0.

                          • -

                        • axis2 (int, optional) – The second axis of the 2-D sub-arrays from which +

                        • axis2 (int, optional) – The second axis of the 2-D sub-arrays from which the diagonals should be taken. Default: 1.

                        • dtype (Dtype, optional) – Data type of the output array. If unspecified the output type is inferred from the input array.

                          • diff --git a/docs/build/html/python/_autosummary/mlx.core.transpose.html b/docs/build/html/python/_autosummary/mlx.core.transpose.html index b25f19398..67dabb84c 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.29.2 documentation + mlx.core.transpose — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                          mlx.core.transpose#

                          -transpose(a: array, /, axes: Sequence[int] | None = None, *, stream: None | Stream | Device = None) → array#
                          +transpose(a: array, /, axes: Sequence[int] | None = None, *, stream: None | Stream | Device = None) → array#

                          Transpose the dimensions of the array.

                          Parameters:

                          • a (array) – Input array.

                            • -

                          • axes (list(int), optional) – Specifies the source axis for each axis +

                          • axes (list(int), optional) – Specifies the source axis for each axis in the new array. The default is to reverse the axes.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.tri.html b/docs/build/html/python/_autosummary/mlx.core.tri.html index e40ab2981..8b21d8d77 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.29.2 documentation + mlx.core.tri — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                          mlx.core.tri#

                          -tri(n: int, m: int, k: int, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array#
                          +tri(n: int, m: int, k: int, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) → array#

                          An array with ones at and below the given diagonal and zeros elsewhere.

                          Parameters:
                            -

                          • n (int) – The number of rows in the output.

                            • -

                          • m (int, optional) – The number of cols in the output. Defaults to None.

                            • -

                          • k (int, optional) – The diagonal of the 2-D array. Defaults to 0.

                            • +

                          • n (int) – The number of rows in the output.

                            • +

                          • m (int, optional) – The number of cols in the output. Defaults to None.

                            • +

                          • k (int, optional) – The diagonal of the 2-D array. Defaults to 0.

                          • dtype (Dtype, optional) – Data type of the output array. Defaults to float32.

                          • stream (Stream, optional) – Stream or device. Defaults to None.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.tril.html b/docs/build/html/python/_autosummary/mlx.core.tril.html index 568816ddc..e6a702e90 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.29.2 documentation + mlx.core.tril — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                          mlx.core.tril#

                          -tril(x: array, k: int, *, stream: None | Stream | Device = None) → array#
                          +tril(x: array, k: int, *, stream: None | Stream | Device = None) → array#

                          Zeros the array above the given diagonal.

                          Parameters:

                          • x (array) – input array.

                            • -

                          • k (int, optional) – The diagonal of the 2-D array. Defaults to 0.

                            • +

                          • k (int, optional) – The diagonal of the 2-D array. Defaults to 0.

                          • stream (Stream, optional) – Stream or device. Defaults to None.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.triu.html b/docs/build/html/python/_autosummary/mlx.core.triu.html index 591dab501..857e8f67e 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.29.2 documentation + mlx.core.triu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                          mlx.core.triu#

                          -triu(x: array, k: int, *, stream: None | Stream | Device = None) → array#
                          +triu(x: array, k: int, *, stream: None | Stream | Device = None) → array#

                          Zeros the array below the given diagonal.

                          Parameters:

                          • x (array) – input array.

                            • -

                          • k (int, optional) – The diagonal of the 2-D array. Defaults to 0.

                            • +

                          • k (int, optional) – The diagonal of the 2-D array. Defaults to 0.

                          • stream (Stream, optional) – Stream or device. Defaults to None.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.unflatten.html b/docs/build/html/python/_autosummary/mlx.core.unflatten.html index c6743947a..e178e2b1d 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.29.2 documentation + mlx.core.unflatten — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                          mlx.core.unflatten#

                          -unflatten(a: array, /, axis: int, shape: Sequence[int], *, stream: None | Stream | Device = None) → array#
                          +unflatten(a: array, /, axis: int, shape: Sequence[int], *, stream: None | Stream | Device = None) → array#

                          Unflatten an axis of an array to a shape.

                          Parameters:

                          • a (array) – Input array.

                            • -

                          • axis (int) – The axis to unflatten.

                            • -

                          • shape (tuple(int)) – The shape to unflatten to. At most one +

                          • axis (int) – The axis to unflatten.

                            • +

                          • shape (tuple(int)) – The shape to unflatten to. At most one entry can be -1 in which case the corresponding size will be inferred.

                          • stream (Stream, optional) – Stream or device. Defaults to None 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 1c004364c..d0667f4b2 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.29.2 documentation + mlx.core.value_and_grad — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                            mlx.core.value_and_grad#

                            -value_and_grad(fun: Callable, argnums: int | Sequence[int] | None = None, argnames: str | Sequence[str] = []) → Callable#
                            +value_and_grad(fun: Callable, argnums: int | Sequence[int] | None = None, argnames: str | Sequence[str] = []) → Callable#

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

                            The function passed to value_and_grad() should return either a scalar loss or a tuple in which the first element is a scalar @@ -966,12 +966,12 @@ loss and the remaining elements can be anything.

                            array or trees of array and returns a scalar output array or a tuple the first element of which should be a scalar array.

                            • -

                          • argnums (int or list(int), optional) – Specify the index (or indices) +

                          • argnums (int or list(int), optional) – Specify the index (or indices) of the positional arguments of fun to compute the gradient with respect to. If neither argnums nor argnames are provided argnums defaults to 0 indicating fun’s first argument.

                            • -

                          • argnames (str or list(str), optional) – Specify keyword arguments of +

                          • argnames (str or list(str), optional) – Specify keyword arguments of fun to compute gradients with respect to. It defaults to [] so no gradients for keyword arguments by default.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.var.html b/docs/build/html/python/_autosummary/mlx.core.var.html index d8f162ba7..57c1add6a 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.29.2 documentation + mlx.core.var — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,18 +932,18 @@ document.write(`

                          mlx.core.var#

                          -var(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, ddof: int = 0, *, stream: None | Stream | Device = None) → array#
                          +var(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, ddof: int = 0, *, stream: None | Stream | Device = None) → array#

                          Compute the variance(s) over the given axes.

                          Parameters:

                          • a (array) – Input array.

                            • -

                          • axis (int or list(int), optional) – Optional axis or +

                          • axis (int or list(int), optional) – Optional axis or axes to reduce over. If unspecified this defaults to reducing over the entire array.

                            • -

                          • keepdims (bool, optional) – Keep reduced axes as +

                          • keepdims (bool, optional) – Keep reduced axes as singleton dimensions, defaults to False.

                            • -

                          • ddof (int, optional) – The divisor to compute the variance +

                          • ddof (int, optional) – The divisor to compute the variance is N - ddof, defaults to 0.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.view.html b/docs/build/html/python/_autosummary/mlx.core.view.html index 6dea85819..e9462c5b8 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.29.2 documentation + mlx.core.view — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.view#

                          -view(a: scalar | array, dtype: Dtype, stream: None | Stream | Device = None) → array#
                          +view(a: scalar | array, dtype: Dtype, stream: None | Stream | Device = None) → array#

                          View the array as a different type.

                          The output shape changes along the last axis if the input array’s type and the input dtype do not have the same size.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.vjp.html b/docs/build/html/python/_autosummary/mlx.core.vjp.html index 5d4018b2f..493513e5b 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.29.2 documentation + mlx.core.vjp — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.vjp#

                          -vjp(fun: Callable, primals: list[array], cotangents: list[array]) → tuple[list[array], list[array]]#
                          +vjp(fun: Callable, primals: list[array], cotangents: list[array]) → tuple[list[array], list[array]]#

                          Compute the vector-Jacobian product.

                          Computes the product of the cotangents with the Jacobian of a function fun evaluated at primals.

                          @@ -941,9 +941,9 @@ function fun

                          • fun (Callable) – A function which takes a variable number of array and returns a single array or list of array.

                            • -

                          • primals (list(array)) – A list of array at which to +

                          • primals (list(array)) – A list of array at which to evaluate the Jacobian.

                            • -

                          • cotangents (list(array)) – A list of array which are the +

                          • cotangents (list(array)) – A list of array which are the “vector” in the vector-Jacobian product. The cotangents should be the same in number, shape, and type as the outputs of fun.

                          @@ -953,7 +953,7 @@ same in number, shape, and type as the outputs of fun.

                          Return type:
                          -

                          list(array)

                          +

                          list(array)

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.vmap.html b/docs/build/html/python/_autosummary/mlx.core.vmap.html index 5753002bd..452831af8 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.29.2 documentation + mlx.core.vmap — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.vmap#

                          -vmap(fun: Callable, in_axes: object = 0, out_axes: object = 0) → Callable#
                          +vmap(fun: Callable, in_axes: object = 0, out_axes: object = 0) → Callable#

                          Returns a vectorized version of fun.

                          Parameters:
                          @@ -940,11 +940,11 @@ document.write(`

                        • fun (Callable) – A function which takes a variable number of array or a tree of array and returns a variable number of array or a tree of array.

                          • -

                        • in_axes (int, optional) – An integer or a valid prefix tree of the +

                        • in_axes (int, optional) – An integer or a valid prefix tree of the inputs to fun where each node specifies the vmapped axis. If the value is None then the corresponding input(s) are not vmapped. Defaults to 0.

                          • -

                        • out_axes (int, optional) – An integer or a valid prefix tree of the +

                        • out_axes (int, optional) – An integer or a valid prefix tree of the outputs of fun where each node specifies the vmapped axis. If the value is None then the corresponding outputs(s) are not vmapped. Defaults to 0.

                          • diff --git a/docs/build/html/python/_autosummary/mlx.core.where.html b/docs/build/html/python/_autosummary/mlx.core.where.html index af73e038a..17f5788a0 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.29.2 documentation + mlx.core.where — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.where#

                          -where(condition: scalar | array, x: scalar | array, y: scalar | array, /, *, stream: None | Stream | Device = None) → array#
                          +where(condition: scalar | array, x: scalar | array, y: scalar | array, /, *, stream: None | Stream | Device = None) → array#

                          Select from x or y according to condition.

                          The condition and input arrays must be the same shape or broadcastable with each another.

                          diff --git a/docs/build/html/python/_autosummary/mlx.core.zeros.html b/docs/build/html/python/_autosummary/mlx.core.zeros.html index caeba2f71..e1f035fed 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.29.2 documentation + mlx.core.zeros — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,12 +932,12 @@ document.write(`

                          mlx.core.zeros#

                          -zeros(shape: int | Sequence[int], dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#
                          +zeros(shape: int | Sequence[int], dtype: Dtype | None = float32, *, stream: None | Stream | Device = None) → array#

                          Construct an array of zeros.

                          Parameters:
                            -

                          • shape (int or list(int)) – The shape of the output array.

                            • +

                          • shape (int or list(int)) – The shape of the output array.

                          • dtype (Dtype, optional) – Data type of the output array. If unspecified the output type defaults to float32.

                          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 de8c23dd5..1484b6211 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.29.2 documentation + mlx.core.zeros_like — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.core.zeros_like#

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

                          An array of zeros like the input.

                          Parameters:
                          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 ed80b2e2d..8d7d18784 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.29.2 documentation + mlx.nn.average_gradients — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.nn.average_gradients#

                          -average_gradients(gradients: Any, group: Group | None = None, all_reduce_size: int = 33554432, communication_type: Dtype | None = None, communication_stream: Stream | None = None)#
                          +average_gradients(gradients: Any, group: Group | None = None, all_reduce_size: int = 33554432, communication_type: Dtype | None = None, communication_stream: Stream | None = None)#

                          Average the gradients across the distributed processes in the passed group.

                          This helper enables concatenating several gradients of small arrays to one big all reduce call for better networking performance.

                          @@ -944,7 +944,7 @@ have the same structure across processes)

                        • group (Optional[Group]) – The group of processes to average the gradients. If set to None the global group is used. Default: None.

                          • -

                        • all_reduce_size (int) – Group arrays until their size in bytes exceeds +

                        • all_reduce_size (int) – Group arrays until their size in bytes exceeds this number. Perform one communication step per group of arrays. If less or equal to 0 array grouping is disabled. Default: 32MiB.

                        • communication_type (Optional[Dtype]) – If provided cast to this diff --git a/docs/build/html/python/_autosummary/mlx.nn.quantize.html b/docs/build/html/python/_autosummary/mlx.nn.quantize.html index f7ac3c02b..60cf22b6c 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.29.2 documentation + mlx.nn.quantize — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.nn.quantize#

                          -quantize(model: Module, group_size: int = 64, bits: int = 4, *, mode: str = 'affine', class_predicate: Callable[[str, Module], bool | dict] | None = None)#
                          +quantize(model: Module, group_size: int = 64, bits: int = 4, *, mode: str = 'affine', class_predicate: Callable[[str, Module], bool | dict] | None = None)#

                          Quantize the sub-modules of a module according to a predicate.

                          By default all layers that define a to_quantized(group_size, bits) method will be quantized. Both Linear and Embedding layers @@ -941,11 +941,11 @@ will be quantized. Note also, the module is updated in-place.

                          Parameters:

                          • model (Module) – The model whose leaf modules may be quantized.

                            • -

                          • group_size (int) – The quantization group size (see +

                          • group_size (int) – The quantization group size (see mlx.core.quantize()). Default: 64.

                            • -

                          • bits (int) – The number of bits per parameter (see +

                          • bits (int) – The number of bits per parameter (see mlx.core.quantize()). Default: 4.

                            • -

                          • mode (str) – The quantization method to use (see +

                          • mode (str) – The quantization method to use (see mlx.core.quantize()). Default: "affine".

                          • class_predicate (Optional[Callable]) – A callable which receives the Module path and Module itself and returns True or a 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 a9c31a46d..f94bc201e 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.29.2 documentation + mlx.nn.value_and_grad — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                            mlx.nn.value_and_grad#

                            -value_and_grad(model: Module, fn: Callable)#
                            +value_and_grad(model: Module, fn: Callable)#

                            Transform the passed function fn to a function that computes the gradients of fn wrt the model’s trainable parameters and also its value.

                            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 d9ced37a5..818fe277a 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.29.2 documentation + mlx.optimizers.clip_grad_norm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -947,8 +947,8 @@ greater than max_no
                            Parameters:
                              -

                            • grads (dict) – A dictionary containing the gradient arrays.

                              • -

                            • max_norm (float) – The maximum allowed global norm of the gradients.

                              • +

                            • grads (dict) – A dictionary containing the gradient arrays.

                              • +

                            • max_norm (float) – The maximum allowed global norm of the gradients.

                            Returns:
                            @@ -956,7 +956,7 @@ greater than max_no gradient norm.

                            Return type:
                            -

                            (dict, float)

                            +

                            (dict, float)

                          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 a34267eed..e1b3ad2c4 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.29.2 documentation + mlx.utils.tree_flatten — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.utils.tree_flatten#

                          -tree_flatten(tree: Any, prefix: str = '', is_leaf: Callable | None = None, destination: List[Tuple[str, Any]] | Dict[str, Any] | None = None) → List[Tuple[str, Any]] | Dict[str, Any]#
                          +tree_flatten(tree: Any, prefix: str = '', is_leaf: Callable | None = None, destination: List[Tuple[str, Any]] | Dict[str, Any] | None = None) → List[Tuple[str, Any]] | Dict[str, Any]#

                          Flattens a Python tree to a list of key, value tuples.

                          The keys are using the dot notation to define trees of arbitrary depth and complexity.

                          @@ -956,11 +956,11 @@ complexity.

                          Parameters:

                          • tree (Any) – The Python tree to be flattened.

                            • -

                          • prefix (str) – A prefix to use for the keys. The first character is +

                          • prefix (str) – A prefix to use for the keys. The first character is always discarded.

                          • is_leaf (callable) – An optional callable that returns True if the passed object is considered a leaf or False otherwise.

                            • -

                          • destination (list or dict, optional) – A list or dictionary to store the +

                          • destination (list or dict, optional) – A list or dictionary to store the flattened tree. If None an empty list will be used. Default: None.

                          @@ -972,7 +972,7 @@ flattened tree. If None an empty list will be used. Default: Return type: -

                          Union[List[Tuple[str, Any]], Dict[str, Any]]

                          +

                          Union[List[Tuple[str, Any]], Dict[str, Any]]

                          • 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 7ba8e563f..f9cb733f8 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.29.2 documentation + mlx.utils.tree_map — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                          mlx.utils.tree_map#

                          -tree_map(fn: Callable, tree: Any, *rest: Any, is_leaf: Callable | None = None) → Any#
                          +tree_map(fn: Callable, tree: Any, *rest: Any, is_leaf: Callable | None = None) → Any#

                          Applies fn to the leaves of the Python tree tree and returns a new collection with the results.

                          If rest is provided, every item is assumed to be a superset of tree and the corresponding leaves are provided as extra positional arguments to -fn. In that respect, tree_map() is closer to itertools.starmap() -than to map().

                          +fn. In that respect, tree_map() is closer to itertools.starmap() +than to map().

                          The keyword argument is_leaf decides what constitutes a leaf from tree similar to tree_flatten().

                          import mlx.nn as nn
@@ -957,7 +957,7 @@ than to 
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 9b021bd93..84e753409 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.29.2 documentation
+    mlx.utils.tree_map_with_path — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                          @@ -932,7 +932,7 @@ document.write(`

                          mlx.utils.tree_map_with_path#

                          -tree_map_with_path(fn: Callable, tree: Any, *rest: Any, is_leaf: Callable | None = None, path: Any | None = None) → Any#
                          +tree_map_with_path(fn: Callable, tree: Any, *rest: Any, is_leaf: Callable | None = None, path: Any | None = None) → Any#

                          Applies fn to the path and leaves of the Python tree tree and returns a new collection with the results.

                          This function is the same tree_map() but the fn takes the path as @@ -942,7 +942,7 @@ the first argument followed by the remaining tree nodes.

                          • fn (callable) – The function that processes the leaves of the tree.

                          • tree (Any) – The main Python tree that will be iterated upon.

                            • -

                          • rest (tuple[Any]) – Extra trees to be iterated together with tree.

                            • +

                          • rest (tuple[Any]) – Extra trees to be iterated together with tree.

                          • is_leaf (Optional[Callable]) – An optional callable that returns True if the passed object is considered a leaf or False otherwise.

                          • path (Optional[Any]) – Prefix will be added to the result.

                            • 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 7fed72db8..f380d1762 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.29.2 documentation + mlx.utils.tree_reduce — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                          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 c244c343c..70da89bbc 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.29.2 documentation + mlx.utils.tree_unflatten — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.utils.tree_unflatten#

                          -tree_unflatten(tree: List[Tuple[str, Any]] | Dict[str, Any]) → Any#
                          +tree_unflatten(tree: List[Tuple[str, Any]] | Dict[str, Any]) → Any#

                          Recreate a Python tree from its flat representation.

                          from mlx.utils import tree_unflatten
 
@@ -947,7 +947,7 @@ document.write(`
                          Parameters:
                          -

                          tree (list[tuple[str, Any]] or dict[str, Any]) – The flat representation of a Python tree. +

                          tree (list[tuple[str, Any]] or dict[str, Any]) – The flat representation of a Python tree. For instance as returned by tree_flatten().

                          Returns:
                          diff --git a/docs/build/html/python/_autosummary/stream_class.html b/docs/build/html/python/_autosummary/stream_class.html index 218057cd5..8874d19a4 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.29.2 documentation + mlx.core.Stream — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                          diff --git a/docs/build/html/python/array.html b/docs/build/html/python/array.html index 2806d1639..f0405fda7 100644 --- a/docs/build/html/python/array.html +++ b/docs/build/html/python/array.html @@ -8,7 +8,7 @@ - Array — MLX 0.29.2 documentation + Array — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -935,7 +935,7 @@ document.write(`

                          Access the value of a scalar array.

                          array.tolist(self)

                          -

                          Convert the array to a Python list.

                          +

                          Convert the array to a Python list.

                          array.dtype

                          The array's Dtype.

                          @@ -1043,7 +1043,7 @@ document.write(`

                          See reciprocal().

                          array.reshape(self, *shape[, stream])

                          -

                          Equivalent to reshape() but the shape can be passed either as a tuple or as separate arguments.

                          +

                          Equivalent to reshape() but the shape can be passed either as a tuple or as separate arguments.

                          array.round(self[, decimals, stream])

                          See round().

                          diff --git a/docs/build/html/python/cuda.html b/docs/build/html/python/cuda.html index 90fdc9d50..6c4564051 100644 --- a/docs/build/html/python/cuda.html +++ b/docs/build/html/python/cuda.html @@ -8,7 +8,7 @@ - CUDA — MLX 0.29.2 documentation + CUDA — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/data_types.html b/docs/build/html/python/data_types.html index e0b399519..bb66a8bc9 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.29.2 documentation + Data Types — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/devices_and_streams.html b/docs/build/html/python/devices_and_streams.html index d24b12302..5fd53dc7c 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.29.2 documentation + Devices and Streams — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/distributed.html b/docs/build/html/python/distributed.html index 172bfeef0..bc58cf644 100644 --- a/docs/build/html/python/distributed.html +++ b/docs/build/html/python/distributed.html @@ -8,7 +8,7 @@ - Distributed Communication — MLX 0.29.2 documentation + Distributed Communication — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/export.html b/docs/build/html/python/export.html index 62ff024bb..2141429d0 100644 --- a/docs/build/html/python/export.html +++ b/docs/build/html/python/export.html @@ -8,7 +8,7 @@ - Export Functions — MLX 0.29.2 documentation + Export Functions — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -922,8 +922,8 @@ document.write(`

                          Export Functions#

                          - - + + diff --git a/docs/build/html/python/fast.html b/docs/build/html/python/fast.html index c7228429e..966ca42f7 100644 --- a/docs/build/html/python/fast.html +++ b/docs/build/html/python/fast.html @@ -8,7 +8,7 @@ - Fast — MLX 0.29.2 documentation + Fast — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/fft.html b/docs/build/html/python/fft.html index 1d7219ccf..389842bbf 100644 --- a/docs/build/html/python/fft.html +++ b/docs/build/html/python/fft.html @@ -8,7 +8,7 @@ - FFT — MLX 0.29.2 documentation + FFT — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/linalg.html b/docs/build/html/python/linalg.html index 4cf721d78..2c1599803 100644 --- a/docs/build/html/python/linalg.html +++ b/docs/build/html/python/linalg.html @@ -8,7 +8,7 @@ - Linear Algebra — MLX 0.29.2 documentation + Linear Algebra — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/memory_management.html b/docs/build/html/python/memory_management.html index dd233a561..abd435f19 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.29.2 documentation + Memory Management — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/metal.html b/docs/build/html/python/metal.html index b3f8ba6be..75afb6778 100644 --- a/docs/build/html/python/metal.html +++ b/docs/build/html/python/metal.html @@ -8,7 +8,7 @@ - Metal — MLX 0.29.2 documentation + Metal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/nn.html b/docs/build/html/python/nn.html index 5f70cd767..b20aec874 100644 --- a/docs/build/html/python/nn.html +++ b/docs/build/html/python/nn.html @@ -8,7 +8,7 @@ - Neural Networks — MLX 0.29.2 documentation + Neural Networks — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 6dabeac0f..965d670e3 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.29.2 documentation + mlx.nn.ALiBi — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 666b18a7f..e6bed672f 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.29.2 documentation + mlx.nn.AvgPool1d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                          mlx.nn.AvgPool1d#

                          -class AvgPool1d(kernel_size: int | Tuple[int], stride: int | Tuple[int] | None = None, padding: int | Tuple[int] = 0)#
                          +class AvgPool1d(kernel_size: int | Tuple[int], stride: int | Tuple[int] | None = None, padding: int | Tuple[int] = 0)#

                          Applies 1-dimensional average pooling.

                          Spatially downsamples the input by taking the average of a sliding window of size kernel_size and sliding stride stride.

                          Parameters:
                            -

                          • kernel_size (int or tuple(int)) – The size of the pooling window kernel.

                            • -

                          • stride (int or tuple(int), optional) – The stride of the pooling window. +

                          • kernel_size (int or tuple(int)) – The size of the pooling window kernel.

                            • +

                          • stride (int or tuple(int), optional) – The stride of the pooling window. Default: kernel_size.

                            • -

                          • padding (int or tuple(int), optional) – How much zero padding to apply to +

                          • padding (int or tuple(int), optional) – How much zero padding to apply to the input. The padding amount is applied to both sides of the spatial axis. Default: 0.

                          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 3b1f9d936..71c5a05c6 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.29.2 documentation + mlx.nn.AvgPool2d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.nn.AvgPool2d#

                          -class AvgPool2d(kernel_size: int | Tuple[int, int], stride: int | Tuple[int, int] | None = None, padding: int | Tuple[int, int] | None = 0)#
                          +class AvgPool2d(kernel_size: int | Tuple[int, int], stride: int | Tuple[int, int] | None = None, padding: int | Tuple[int, int] | None = 0)#

                          Applies 2-dimensional average pooling.

                          Spatially downsamples the input by taking the average of a sliding window of size kernel_size and sliding stride stride.

                          @@ -946,10 +946,10 @@ used for the height axis, the second
                          Parameters:
                            -

                          • kernel_size (int or tuple(int, int)) – The size of the pooling window.

                            • -

                          • stride (int or tuple(int, int), optional) – The stride of the pooling +

                          • kernel_size (int or tuple(int, int)) – The size of the pooling window.

                            • +

                          • stride (int or tuple(int, int), optional) – The stride of the pooling window. Default: kernel_size.

                            • -

                          • padding (int or tuple(int, int), optional) – How much zero +

                          • padding (int or tuple(int, int), optional) – How much zero padding to apply to the input. The padding is applied on both sides of the height and width axis. Default: 0.

                          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 b32fecd14..b91d73b99 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.29.2 documentation + mlx.nn.AvgPool3d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.nn.AvgPool3d#

                          -class AvgPool3d(kernel_size: int | Tuple[int, int, int], stride: int | Tuple[int, int, int] | None = None, padding: int | Tuple[int, int, int] | None = 0)#
                          +class AvgPool3d(kernel_size: int | Tuple[int, int, int], stride: int | Tuple[int, int, int] | None = None, padding: int | Tuple[int, int, int] | None = 0)#

                          Applies 3-dimensional average pooling.

                          Spatially downsamples the input by taking the average of a sliding window of size kernel_size and sliding stride stride.

                          @@ -947,10 +947,10 @@ for the depth axis, the second
                          Parameters:
                            -

                          • kernel_size (int or tuple(int, int, int)) – The size of the pooling window.

                            • -

                          • stride (int or tuple(int, int, int), optional) – The stride of the pooling +

                          • kernel_size (int or tuple(int, int, int)) – The size of the pooling window.

                            • +

                          • stride (int or tuple(int, int, int), optional) – The stride of the pooling window. Default: kernel_size.

                            • -

                          • padding (int or tuple(int, int, int), optional) – How much zero +

                          • padding (int or tuple(int, int, int), optional) – How much zero padding to apply to the input. The padding is applied on both sides of the depth, height and width axis. Default: 0.

                          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 caa297277..dc49b93d5 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.29.2 documentation + mlx.nn.BatchNorm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                          mlx.nn.BatchNorm#

                          -class BatchNorm(num_features: int, eps: float = 1e-05, momentum: float = 0.1, affine: bool = True, track_running_stats: bool = True)#
                          +class BatchNorm(num_features: int, eps: float = 1e-05, momentum: float = 0.1, affine: bool = True, track_running_stats: bool = True)#

                          Applies Batch Normalization over a 2D or 3D input.

                          Computes

                          @@ -951,14 +951,14 @@ Covariate Shift.

                          Parameters:
                            -

                          • num_features (int) – The feature dimension to normalize over.

                            • -

                          • eps (float, optional) – A small additive constant for numerical +

                          • num_features (int) – The feature dimension to normalize over.

                            • +

                          • eps (float, optional) – A small additive constant for numerical stability. Default: 1e-5.

                            • -

                          • momentum (float, optional) – The momentum for updating the running +

                          • momentum (float, optional) – The momentum for updating the running mean and variance. Default: 0.1.

                            • -

                          • affine (bool, optional) – If True, apply a learned affine +

                          • affine (bool, optional) – If True, apply a learned affine transformation after the normalization. Default: True.

                            • -

                          • track_running_stats (bool, optional) – If True, track the +

                          • track_running_stats (bool, optional) – If True, track the running mean and variance. Default: True.

                          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 6700c21a5..ca706b434 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.29.2 documentation + mlx.nn.CELU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                          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 a9272c259..b93948103 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.29.2 documentation + mlx.nn.Conv1d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                          mlx.nn.Conv1d#

                          -class Conv1d(in_channels: int, out_channels: int, kernel_size: int, stride: int = 1, padding: int = 0, dilation: int = 1, groups: int = 1, bias: bool = True)#
                          +class Conv1d(in_channels: int, out_channels: int, kernel_size: int, stride: int = 1, padding: int = 0, dilation: int = 1, groups: int = 1, bias: bool = True)#

                          Applies a 1-dimensional convolution over the multi-channel input sequence.

                          The channels are expected to be last i.e. the input shape should be NLC where:

                            @@ -943,17 +943,17 @@ document.write(`
                            Parameters:
                              -

                            • in_channels (int) – The number of input channels

                              • -

                            • out_channels (int) – The number of output channels

                              • -

                            • kernel_size (int) – The size of the convolution filters

                              • -

                            • stride (int, optional) – The stride when applying the filter. +

                            • in_channels (int) – The number of input channels

                              • +

                            • out_channels (int) – The number of output channels

                              • +

                            • kernel_size (int) – The size of the convolution filters

                              • +

                            • stride (int, optional) – The stride when applying the filter. Default: 1.

                              • -

                            • padding (int, optional) – How many positions to 0-pad the input with. +

                            • padding (int, optional) – How many positions to 0-pad the input with. Default: 0.

                              • -

                            • dilation (int, optional) – The dilation of the convolution.

                              • -

                            • groups (int, optional) – The number of groups for the convolution. +

                            • dilation (int, optional) – The dilation of the convolution.

                              • +

                            • groups (int, optional) – The number of groups for the convolution. Default: 1.

                              • -

                            • bias (bool, optional) – If True add a learnable bias to the output. +

                            • bias (bool, optional) – If True add a learnable bias to the output. Default: True

                            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 ba90dd63b..7cdc3652d 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.29.2 documentation + mlx.nn.Conv2d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                            mlx.nn.Conv2d#

                            -class Conv2d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, groups: int = 1, bias: bool = True)#
                            +class Conv2d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, groups: int = 1, bias: bool = True)#

                            Applies a 2-dimensional convolution over the multi-channel input image.

                            The channels are expected to be last i.e. the input shape should be NHWC where:

                              @@ -944,17 +944,17 @@ document.write(`
                              Parameters:
                                -

                              • in_channels (int) – The number of input channels.

                                • -

                              • out_channels (int) – The number of output channels.

                                • -

                              • kernel_size (int or tuple) – The size of the convolution filters.

                                • -

                              • stride (int or tuple, optional) – The size of the stride when +

                              • in_channels (int) – The number of input channels.

                                • +

                              • out_channels (int) – The number of output channels.

                                • +

                              • kernel_size (int or tuple) – The size of the convolution filters.

                                • +

                              • stride (int or tuple, optional) – The size of the stride when applying the filter. Default: 1.

                                • -

                              • padding (int or tuple, optional) – How many positions to 0-pad +

                              • padding (int or tuple, optional) – How many positions to 0-pad the input with. Default: 0.

                                • -

                              • dilation (int or tuple, optional) – The dilation of the convolution.

                                • -

                              • groups (int, optional) – The number of groups for the convolution. +

                              • dilation (int or tuple, optional) – The dilation of the convolution.

                                • +

                              • groups (int, optional) – The number of groups for the convolution. Default: 1.

                                • -

                              • bias (bool, optional) – If True add a learnable bias to the +

                              • bias (bool, optional) – If True add a learnable bias to the output. Default: True

                              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 3a2a1e0fc..5bb66db8f 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.29.2 documentation + mlx.nn.Conv3d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                              mlx.nn.Conv3d#

                              -class Conv3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, bias: bool = True)#
                              +class Conv3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, bias: bool = True)#

                              Applies a 3-dimensional convolution over the multi-channel input image.

                              The channels are expected to be last i.e. the input shape should be NDHWC where:

                                @@ -945,15 +945,15 @@ document.write(`
                                Parameters:
                                  -

                                • in_channels (int) – The number of input channels.

                                  • -

                                • out_channels (int) – The number of output channels.

                                  • -

                                • kernel_size (int or tuple) – The size of the convolution filters.

                                  • -

                                • stride (int or tuple, optional) – The size of the stride when +

                                • in_channels (int) – The number of input channels.

                                  • +

                                • out_channels (int) – The number of output channels.

                                  • +

                                • kernel_size (int or tuple) – The size of the convolution filters.

                                  • +

                                • stride (int or tuple, optional) – The size of the stride when applying the filter. Default: 1.

                                  • -

                                • dilation (int or tuple, optional) – The dilation of the convolution.

                                  • -

                                • padding (int or tuple, optional) – How many positions to 0-pad +

                                • dilation (int or tuple, optional) – The dilation of the convolution.

                                  • +

                                • padding (int or tuple, optional) – How many positions to 0-pad the input with. Default: 0.

                                  • -

                                • bias (bool, optional) – If True add a learnable bias to the +

                                • bias (bool, optional) – If True add a learnable bias to the output. Default: True

                                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 ee5a16c34..5a623917f 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.29.2 documentation + mlx.nn.ConvTranspose1d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                mlx.nn.ConvTranspose1d#

                                -class ConvTranspose1d(in_channels: int, out_channels: int, kernel_size: int, stride: int = 1, padding: int = 0, dilation: int = 1, output_padding: int = 0, bias: bool = True)#
                                +class ConvTranspose1d(in_channels: int, out_channels: int, kernel_size: int, stride: int = 1, padding: int = 0, dilation: int = 1, output_padding: int = 0, bias: bool = True)#

                                Applies a 1-dimensional transposed convolution over the multi-channel input sequence.

                                The channels are expected to be last i.e. the input shape should be NLC where:

                                  @@ -943,17 +943,17 @@ document.write(`
                                  Parameters:
                                    -

                                  • in_channels (int) – The number of input channels

                                    • -

                                  • out_channels (int) – The number of output channels

                                    • -

                                  • kernel_size (int) – The size of the convolution filters

                                    • -

                                  • stride (int, optional) – The stride when applying the filter. +

                                  • in_channels (int) – The number of input channels

                                    • +

                                  • out_channels (int) – The number of output channels

                                    • +

                                  • kernel_size (int) – The size of the convolution filters

                                    • +

                                  • stride (int, optional) – The stride when applying the filter. Default: 1.

                                    • -

                                  • padding (int, optional) – How many positions to 0-pad the input with. +

                                  • padding (int, optional) – How many positions to 0-pad the input with. Default: 0.

                                    • -

                                  • dilation (int, optional) – The dilation of the convolution.

                                    • -

                                  • output_padding (int, optional) – Additional size added to one side of the +

                                  • dilation (int, optional) – The dilation of the convolution.

                                    • +

                                  • output_padding (int, optional) – Additional size added to one side of the output shape. Default: 0.

                                    • -

                                  • bias (bool, optional) – If True add a learnable bias to the output. +

                                  • bias (bool, optional) – If True add a learnable bias to the output. Default: True

                                  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 959aa680f..90bb108a7 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.29.2 documentation + mlx.nn.ConvTranspose2d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                  mlx.nn.ConvTranspose2d#

                                  -class ConvTranspose2d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, output_padding: int | tuple = 0, bias: bool = True)#
                                  +class ConvTranspose2d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, output_padding: int | tuple = 0, bias: bool = True)#

                                  Applies a 2-dimensional transposed convolution over the multi-channel input image.

                                  The channels are expected to be last i.e. the input shape should be NHWC where:

                                    @@ -944,17 +944,17 @@ document.write(`
                                    Parameters:
                                      -

                                    • in_channels (int) – The number of input channels.

                                      • -

                                    • out_channels (int) – The number of output channels.

                                      • -

                                    • kernel_size (int or tuple) – The size of the convolution filters.

                                      • -

                                    • stride (int or tuple, optional) – The size of the stride when +

                                    • in_channels (int) – The number of input channels.

                                      • +

                                    • out_channels (int) – The number of output channels.

                                      • +

                                    • kernel_size (int or tuple) – The size of the convolution filters.

                                      • +

                                    • stride (int or tuple, optional) – The size of the stride when applying the filter. Default: 1.

                                      • -

                                    • padding (int or tuple, optional) – How many positions to 0-pad +

                                    • padding (int or tuple, optional) – How many positions to 0-pad the input with. Default: 0.

                                      • -

                                    • dilation (int or tuple, optional) – The dilation of the convolution.

                                      • -

                                    • output_padding (int or tuple, optional) – Additional size added to one +

                                    • dilation (int or tuple, optional) – The dilation of the convolution.

                                      • +

                                    • output_padding (int or tuple, optional) – Additional size added to one side of the output shape. Default: 0.

                                      • -

                                    • bias (bool, optional) – If True add a learnable bias to the +

                                    • bias (bool, optional) – If True add a learnable bias to the output. Default: True

                                    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 d2514070b..fd0c8bed0 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.29.2 documentation + mlx.nn.ConvTranspose3d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                    mlx.nn.ConvTranspose3d#

                                    -class ConvTranspose3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, output_padding: int | tuple = 0, bias: bool = True)#
                                    +class ConvTranspose3d(in_channels: int, out_channels: int, kernel_size: int | tuple, stride: int | tuple = 1, padding: int | tuple = 0, dilation: int | tuple = 1, output_padding: int | tuple = 0, bias: bool = True)#

                                    Applies a 3-dimensional transposed convolution over the multi-channel input image.

                                    The channels are expected to be last i.e. the input shape should be NDHWC where:

                                      @@ -945,17 +945,17 @@ document.write(`
                                      Parameters:
                                        -

                                      • in_channels (int) – The number of input channels.

                                        • -

                                      • out_channels (int) – The number of output channels.

                                        • -

                                      • kernel_size (int or tuple) – The size of the convolution filters.

                                        • -

                                      • stride (int or tuple, optional) – The size of the stride when +

                                      • in_channels (int) – The number of input channels.

                                        • +

                                      • out_channels (int) – The number of output channels.

                                        • +

                                      • kernel_size (int or tuple) – The size of the convolution filters.

                                        • +

                                      • stride (int or tuple, optional) – The size of the stride when applying the filter. Default: 1.

                                        • -

                                      • padding (int or tuple, optional) – How many positions to 0-pad +

                                      • padding (int or tuple, optional) – How many positions to 0-pad the input with. Default: 0.

                                        • -

                                      • dilation (int or tuple, optional) – The dilation of the convolution.

                                        • -

                                      • output_padding (int or tuple, optional) – Additional size added to one +

                                      • dilation (int or tuple, optional) – The dilation of the convolution.

                                        • +

                                      • output_padding (int or tuple, optional) – Additional size added to one side of the output shape. Default: 0.

                                        • -

                                      • bias (bool, optional) – If True add a learnable bias to the +

                                      • bias (bool, optional) – If True add a learnable bias to the output. Default: True

                                      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 ffb2bb035..fc4b6f29b 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.29.2 documentation + mlx.nn.Dropout — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,14 +933,14 @@ document.write(`

                                      mlx.nn.Dropout#

                                      -class Dropout(p: float = 0.5)#
                                      +class Dropout(p: float = 0.5)#

                                      Randomly zero a portion of the elements during training.

                                      The remaining elements are multiplied with \(\frac{1}{1-p}\) where \(p\) is the probability of zeroing an element. This is done so the expected value of a given element will remain the same.

                                      Parameters:
                                      -

                                      p (float) – The probability to zero an element

                                      +

                                      p (float) – The probability to zero an element

                                      Methods

                                      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 837f1076e..10a2541cb 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.29.2 documentation + mlx.nn.Dropout2d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.nn.Dropout2d#

                                      -class Dropout2d(p: float = 0.5)#
                                      +class Dropout2d(p: float = 0.5)#

                                      Apply 2D channel-wise dropout during training.

                                      Randomly zero out entire channels independently with probability \(p\). This layer expects the channels to be last, i.e. the input shape should be @@ -950,7 +950,7 @@ regularize activations. For more details, see [1].

                                      Efficient Object Localization Using Convolutional Networks. CVPR 2015.

                                      Parameters:
                                      -

                                      p (float) – Probability of zeroing a channel during training.

                                      +

                                      p (float) – Probability of zeroing a channel during training.

                                      Methods

                                      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 219d3786f..e4a55c3da 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.29.2 documentation + mlx.nn.Dropout3d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.nn.Dropout3d#

                                      -class Dropout3d(p: float = 0.5)#
                                      +class Dropout3d(p: float = 0.5)#

                                      Apply 3D channel-wise dropout during training.

                                      Randomly zero out entire channels independently with probability \(p\). This layer expects the channels to be last, i.e., the input shape should be @@ -947,7 +947,7 @@ often beneficial for convolutional layers processing 3D data, like in medical imaging or video processing.

                                      Parameters:
                                      -

                                      p (float) – Probability of zeroing a channel during training.

                                      +

                                      p (float) – Probability of zeroing a channel during training.

                                      Methods

                                      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 b62cc995c..15d4340d0 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.29.2 documentation + mlx.nn.ELU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 c614004c1..153f40448 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.29.2 documentation + mlx.nn.Embedding — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,16 +932,16 @@ document.write(`

                                      mlx.nn.Embedding#

                                      -class Embedding(num_embeddings: int, dims: int)#
                                      +class Embedding(num_embeddings: int, dims: int)#

                                      Implements a simple lookup table that maps each input integer to a high-dimensional vector.

                                      Typically used to embed discrete tokens for processing by neural networks.

                                      Parameters:
                                        -

                                      • num_embeddings (int) – How many possible discrete tokens can we embed. +

                                      • num_embeddings (int) – How many possible discrete tokens can we embed. Usually called the vocabulary size.

                                        • -

                                      • dims (int) – The dimensionality of the embeddings.

                                        • +

                                      • dims (int) – The dimensionality of the embeddings.

                                      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 d8ae17096..fc4b35b50 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.29.2 documentation + mlx.nn.GELU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 4fb683bc0..f4aa47406 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.29.2 documentation + mlx.nn.GLU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.nn.GLU#

                                      -class GLU(axis: int = -1)#
                                      +class GLU(axis: int = -1)#

                                      Applies the gated linear unit function.

                                      This function splits the axis dimension of the input into two halves (\(a\) and \(b\)) and applies \(a * \sigma(b)\).

                                      @@ -941,7 +941,7 @@ document.write(` \[\textrm{GLU}(x) = a * \sigma(b)\]
                                      Parameters:
                                      -

                                      axis (int) – The dimension to split along. Default: -1

                                      +

                                      axis (int) – The dimension to split along. Default: -1

                                      Methods

                                      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 0888b0344..b83e1c1f1 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.29.2 documentation + mlx.nn.GRU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.nn.GRU#

                                      -class GRU(input_size: int, hidden_size: int, bias: bool = True)#
                                      +class GRU(input_size: int, hidden_size: int, bias: bool = True)#

                                      A gated recurrent unit (GRU) RNN layer.

                                      The input has shape NLD or LD where:

                                        @@ -955,9 +955,9 @@ time step of shape
                                        Parameters:
                                          -

                                        • input_size (int) – Dimension of the input, D.

                                          • -

                                        • hidden_size (int) – Dimension of the hidden state, H.

                                          • -

                                        • bias (bool) – Whether to use biases or not. Default: True.

                                          • +

                                        • input_size (int) – Dimension of the input, D.

                                          • +

                                        • hidden_size (int) – Dimension of the hidden state, H.

                                          • +

                                        • bias (bool) – Whether to use biases or not. Default: True.

                                        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 5e8858a81..46fe0c99c 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.29.2 documentation + mlx.nn.GroupNorm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                        mlx.nn.GroupNorm#

                                        -class GroupNorm(num_groups: int, dims: int, eps: float = 1e-05, affine: bool = True, pytorch_compatible: bool = False)#
                                        +class GroupNorm(num_groups: int, dims: int, eps: float = 1e-05, affine: bool = True, pytorch_compatible: bool = False)#

                                        Applies Group Normalization [1] to the inputs.

                                        Computes the same normalization as layer norm, namely

                                        @@ -949,12 +949,12 @@ that precede it (except the first) are considered the spatial dimensions.

                                        Parameters:
                                          -

                                        • num_groups (int) – Number of groups to separate the features into

                                          • -

                                        • dims (int) – The feature dimensions of the input to normalize over

                                          • -

                                        • eps (float) – A small additive constant for numerical stability

                                          • -

                                        • affine (bool) – If True learn an affine transform to apply after the +

                                        • num_groups (int) – Number of groups to separate the features into

                                          • +

                                        • dims (int) – The feature dimensions of the input to normalize over

                                          • +

                                        • eps (float) – A small additive constant for numerical stability

                                          • +

                                        • affine (bool) – If True learn an affine transform to apply after the normalization.

                                          • -

                                        • pytorch_compatible (bool) – If True perform the group normalization in +

                                        • pytorch_compatible (bool) – If True perform the group normalization in the same order/grouping as PyTorch.

                                        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 826c23d04..8dd93cd9e 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.29.2 documentation + mlx.nn.HardShrink — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                                        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 7c56abee3..3a6336aa2 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.29.2 documentation + mlx.nn.HardTanh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 6e5fc9b92..1621d5439 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.29.2 documentation + mlx.nn.Hardswish — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 3d486e6d9..668ef02e7 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.29.2 documentation + mlx.nn.InstanceNorm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                        mlx.nn.InstanceNorm#

                                        -class InstanceNorm(dims: int, eps: float = 1e-05, affine: bool = False)#
                                        +class InstanceNorm(dims: int, eps: float = 1e-05, affine: bool = False)#

                                        Applies instance normalization [1] on the inputs.

                                        Computes

                                        @@ -944,9 +944,9 @@ if
                                        Parameters:
                                          -

                                        • dims (int) – The number of features of the input.

                                          • -

                                        • eps (float) – A value added to the denominator for numerical stability. Default: 1e-5.

                                          • -

                                        • affine (bool) – Default: False.

                                          • +

                                        • dims (int) – The number of features of the input.

                                          • +

                                        • eps (float) – A value added to the denominator for numerical stability. Default: 1e-5.

                                          • +

                                        • affine (bool) – Default: False.

                                        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 692f5c360..d7b896004 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.29.2 documentation + mlx.nn.LSTM — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                                        @@ -933,7 +933,7 @@ document.write(`

                                        mlx.nn.LSTM#

                                        -class LSTM(input_size: int, hidden_size: int, bias: bool = True)#
                                        +class LSTM(input_size: int, hidden_size: int, bias: bool = True)#

                                        An LSTM recurrent layer.

                                        The input has shape NLD or LD where:

                                          @@ -958,9 +958,9 @@ each time step, both of shape
                                          Parameters:
                                            -

                                          • input_size (int) – Dimension of the input, D.

                                            • -

                                          • hidden_size (int) – Dimension of the hidden state, H.

                                            • -

                                          • bias (bool) – Whether to use biases or not. Default: True.

                                            • +

                                          • input_size (int) – Dimension of the input, D.

                                            • +

                                          • hidden_size (int) – Dimension of the hidden state, H.

                                            • +

                                          • bias (bool) – Whether to use biases or not. Default: True.

                                        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 9ecad2e99..1af2f3943 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.29.2 documentation + mlx.nn.LayerNorm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                        mlx.nn.LayerNorm#

                                        -class LayerNorm(dims: int, eps: float = 1e-05, affine: bool = True, bias: bool = True)#
                                        +class LayerNorm(dims: int, eps: float = 1e-05, affine: bool = True, bias: bool = True)#

                                        Applies layer normalization [1] on the inputs.

                                        Computes

                                        @@ -944,11 +944,11 @@ parameters initialized at 1 and 0 respectively.

                                        Parameters:
                                          -

                                        • dims (int) – The feature dimension of the input to normalize over

                                          • -

                                        • eps (float) – A small additive constant for numerical stability

                                          • -

                                        • affine (bool) – If True learn an affine transform to apply after the +

                                        • dims (int) – The feature dimension of the input to normalize over

                                          • +

                                        • eps (float) – A small additive constant for numerical stability

                                          • +

                                        • affine (bool) – If True learn an affine transform to apply after the normalization

                                          • -

                                        • bias (bool) – If True include a translation to the affine +

                                        • bias (bool) – If True include a translation to the affine transformation. If set to False the transformation is not really affine just scaling.

                                        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 a8aec5d0a..ca5b80468 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.29.2 documentation + mlx.nn.LeakyReLU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                                        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 a0a2d2e11..c9ab31156 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.29.2 documentation + mlx.nn.Linear — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                        mlx.nn.Linear#

                                        -class Linear(input_dims: int, output_dims: int, bias: bool = True)#
                                        +class Linear(input_dims: int, output_dims: int, bias: bool = True)#

                                        Applies an affine transformation to the input.

                                        Concretely:

                                        @@ -945,9 +945,9 @@ where \(k = \frac{1}{\sqrt{D_i}}\)
                                        Parameters:
                                          -

                                        • input_dims (int) – The dimensionality of the input features

                                          • -

                                        • output_dims (int) – The dimensionality of the output features

                                          • -

                                        • bias (bool, optional) – If set to False then the layer will +

                                        • input_dims (int) – The dimensionality of the input features

                                          • +

                                        • output_dims (int) – The dimensionality of the output features

                                          • +

                                        • bias (bool, optional) – If set to False then the layer will not use a bias. Default is True.

                                        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 aac33d8d9..6aadd900b 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.29.2 documentation + mlx.nn.LogSigmoid — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 4efd09ffc..b42841067 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.29.2 documentation + mlx.nn.LogSoftmax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 3ef13d164..7192e29e5 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.29.2 documentation + mlx.nn.MaxPool1d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,17 +932,17 @@ document.write(`

                                        mlx.nn.MaxPool1d#

                                        -class MaxPool1d(kernel_size: int | Tuple[int], stride: int | Tuple[int] | None = None, padding: int | Tuple[int] = 0)#
                                        +class MaxPool1d(kernel_size: int | Tuple[int], stride: int | Tuple[int] | None = None, padding: int | Tuple[int] = 0)#

                                        Applies 1-dimensional max pooling.

                                        Spatially downsamples the input by taking the maximum of a sliding window of size kernel_size and sliding stride stride.

                                        Parameters:
                                          -

                                        • kernel_size (int or tuple(int)) – The size of the pooling window kernel.

                                          • -

                                        • stride (int or tuple(int), optional) – The stride of the pooling window. +

                                        • kernel_size (int or tuple(int)) – The size of the pooling window kernel.

                                          • +

                                        • stride (int or tuple(int), optional) – The stride of the pooling window. Default: kernel_size.

                                          • -

                                        • padding (int or tuple(int), optional) – How much negative infinity +

                                        • padding (int or tuple(int), optional) – How much negative infinity padding to apply to the input. The padding amount is applied to both sides of the spatial axis. Default: 0.

                                        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 81b564b36..3b47c0d77 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.29.2 documentation + mlx.nn.MaxPool2d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                        mlx.nn.MaxPool2d#

                                        -class MaxPool2d(kernel_size: int | Tuple[int, int], stride: int | Tuple[int, int] | None = None, padding: int | Tuple[int, int] | None = 0)#
                                        +class MaxPool2d(kernel_size: int | Tuple[int, int], stride: int | Tuple[int, int] | None = None, padding: int | Tuple[int, int] | None = 0)#

                                        Applies 2-dimensional max pooling.

                                        Spatially downsamples the input by taking the maximum of a sliding window of size kernel_size and sliding stride stride.

                                        @@ -946,10 +946,10 @@ used for the height axis, the second
                                        Parameters:
                                          -

                                        • kernel_size (int or tuple(int, int)) – The size of the pooling window.

                                          • -

                                        • stride (int or tuple(int, int), optional) – The stride of the pooling +

                                        • kernel_size (int or tuple(int, int)) – The size of the pooling window.

                                          • +

                                        • stride (int or tuple(int, int), optional) – The stride of the pooling window. Default: kernel_size.

                                          • -

                                        • padding (int or tuple(int, int), optional) – How much negative infinity +

                                        • padding (int or tuple(int, int), optional) – How much negative infinity padding to apply to the input. The padding is applied on both sides of the height and width axis. Default: 0.

                                        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 957364f4c..93f6f3862 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.29.2 documentation + mlx.nn.MaxPool3d — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                        mlx.nn.MaxPool3d#

                                        -class MaxPool3d(kernel_size: int | Tuple[int, int, int], stride: int | Tuple[int, int, int] | None = None, padding: int | Tuple[int, int, int] | None = 0)#
                                        +class MaxPool3d(kernel_size: int | Tuple[int, int, int], stride: int | Tuple[int, int, int] | None = None, padding: int | Tuple[int, int, int] | None = 0)#

                                        Applies 3-dimensional max pooling.

                                        Spatially downsamples the input by taking the maximum of a sliding window of size kernel_size and sliding stride stride.

                                        @@ -947,10 +947,10 @@ for the depth axis, the second
                                        Parameters:
                                          -

                                        • kernel_size (int or tuple(int, int, int)) – The size of the pooling window.

                                          • -

                                        • stride (int or tuple(int, int, int), optional) – The stride of the pooling +

                                        • kernel_size (int or tuple(int, int, int)) – The size of the pooling window.

                                          • +

                                        • stride (int or tuple(int, int, int), optional) – The stride of the pooling window. Default: kernel_size.

                                          • -

                                        • padding (int or tuple(int, int, int), optional) – How much negative infinity +

                                        • padding (int or tuple(int, int, int), optional) – How much negative infinity padding to apply to the input. The padding is applied on both sides of the depth, height and width axis. Default: 0.

                                        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 d029613f9..32c42b611 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.29.2 documentation + mlx.nn.Mish — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 20824433d..499686a8e 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.29.2 documentation + mlx.nn.Module.apply — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                        mlx.nn.Module.apply#

                                        -Module.apply(map_fn: Callable[[array], array], filter_fn: Callable[[Module, str, Any], bool] | None = None) → Module#
                                        +Module.apply(map_fn: Callable[[array], array], filter_fn: Callable[[Module, str, Any], bool] | None = None) → Module#

                                        Map all the parameters using the provided map_fn and immediately update the module with the mapped parameters.

                                        For instance running model.apply(lambda x: x.astype(mx.float16)) 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 dde019190..9a4d6707a 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.29.2 documentation + mlx.nn.Module.apply_to_modules — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                        mlx.nn.Module.apply_to_modules#

                                        -Module.apply_to_modules(apply_fn: Callable[[str, Module], Any]) → Module#
                                        +Module.apply_to_modules(apply_fn: Callable[[str, Module], Any]) → Module#

                                        Apply a function to all the modules in this instance (including this instance).

                                        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 531ee0dbb..c87c04551 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.29.2 documentation + mlx.nn.Module.children — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 cf60b5173..45b734d02 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.29.2 documentation + mlx.nn.Module.eval — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 580e26fa1..3f02e563b 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.29.2 documentation + mlx.nn.Module.filter_and_map — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                        mlx.nn.Module.filter_and_map#

                                        -Module.filter_and_map(filter_fn: Callable[[Module, str, Any], bool], map_fn: Callable | None = None, is_leaf_fn: Callable[[Module, str, Any], bool] | None = None)#
                                        +Module.filter_and_map(filter_fn: Callable[[Module, str, Any], bool], map_fn: Callable | None = None, is_leaf_fn: Callable[[Module, str, Any], bool] | None = None)#

                                        Recursively filter the contents of the module using filter_fn, namely only select keys and values where filter_fn returns true.

                                        This is used to implement parameters() and trainable_parameters() 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 2ab9bda00..b3225c74b 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.29.2 documentation + mlx.nn.Module.freeze — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                        mlx.nn.Module.freeze#

                                        -Module.freeze(*, recurse: bool = True, keys: str | List[str] | None = None, strict: bool = False) → Module#
                                        +Module.freeze(*, recurse: bool = True, keys: str | List[str] | None = None, strict: bool = False) → Module#

                                        Freeze the Module’s parameters or some of them. Freezing a parameter means not computing gradients for it.

                                        This function is idempotent i.e. freezing a frozen model is a no-op.

                                        @@ -946,13 +946,13 @@ computing gradients for it.

                                        Parameters:
                                          -

                                        • recurse (bool, optional) – If True then freeze the parameters of the +

                                        • recurse (bool, optional) – If True then freeze the parameters of the submodules as well. Default: True.

                                          • -

                                        • keys (str or list[str], optional) – If provided then only these +

                                        • keys (str or list[str], optional) – If provided then only these parameters will be frozen otherwise all the parameters of a module. For instance freeze all biases by calling module.freeze(keys="bias").

                                          • -

                                        • strict (bool, optional) – If set to True validate that the passed keys exist. +

                                        • strict (bool, optional) – If set to True validate that the passed keys exist. Default: False.

                                        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 586856d7f..e9cf659a7 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.29.2 documentation + mlx.nn.Module.leaf_modules — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 2cfa40301..e845bd82c 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.29.2 documentation + mlx.nn.Module.load_weights — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

                                        mlx.nn.Module.load_weights#

                                        -Module.load_weights(file_or_weights: str | List[Tuple[str, array]], strict: bool = True) → Module#
                                        +Module.load_weights(file_or_weights: str | List[Tuple[str, array]], strict: bool = True) → Module#

                                        Update the model’s weights from a .npz, a .safetensors file, or a list.

                                        Parameters:
                                          -

                                        • file_or_weights (str or list(tuple(str, mx.array))) – The path to +

                                        • file_or_weights (str or list(tuple(str, mx.array))) – The path to the weights .npz file (.npz or .safetensors) or a list of pairs of parameter names and arrays.

                                          • -

                                        • strict (bool, optional) – If True then checks that the provided +

                                        • strict (bool, optional) – If True then checks that the provided weights exactly match the parameters of the model. Otherwise, only the weights actually contained in the model are loaded and shapes are not checked. Default: True.

                                          • 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 f24f7a86e..58e99a507 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.29.2 documentation + mlx.nn.Module.modules — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 e1fb2b4f5..886a1627e 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.29.2 documentation + mlx.nn.Module.named_modules — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 415fb0fcd..df200dbc2 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.29.2 documentation + mlx.nn.Module.parameters — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 e88b506e2..ae36f96d7 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.29.2 documentation + mlx.nn.Module.save_weights — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                          mlx.nn.Module.save_weights#

                                          -Module.save_weights(file: str)#
                                          +Module.save_weights(file: str)#

                                          Save the model’s weights to a file. The saving method is determined by the file extension: - .npz will use mx.savez() - .safetensors will use mx.save_safetensors()

                                          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 d39c8e613..3c720b490 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.29.2 documentation + mlx.nn.Module.set_dtype — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -938,7 +938,7 @@ document.write(`
                                          Parameters:

                                          • dtype (Dtype) – The new dtype.

                                            • -

                                          • predicate (Callable, optional) – A predicate to select +

                                          • predicate (Callable, optional) – A predicate to select parameters to cast. By default, only parameters of type floating will be updated to avoid casting integer parameters to the new dtype.

                                            • 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 3dfef8dbd..e7894dc0c 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.29.2 documentation + mlx.nn.Module.state — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 f98324d0e..820155bfc 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.29.2 documentation + mlx.nn.Module.train — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                                            mlx.nn.Module.train#

                                            -Module.train(mode: bool = True) → Module#
                                            +Module.train(mode: bool = True) → Module#

                                            Set the model in or out of training mode.

                                            Training mode only applies to certain layers. For example Dropout applies a random mask in training mode, but is the identity in evaluation mode.

                                            Parameters:
                                            -

                                            mode (bool) – Indicate if the model should be in training or +

                                            mode (bool) – Indicate if the model should be in training or evaluation mode. Default: True.

                                            Returns:
                                            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 8bd20dcb6..9ddee1c91 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.29.2 documentation + mlx.nn.Module.trainable_parameters — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 7521850d5..a9043f8b0 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.29.2 documentation + mlx.nn.Module.training — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 0bb1ff086..2c1703045 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.29.2 documentation + mlx.nn.Module.unfreeze — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                            mlx.nn.Module.unfreeze#

                                            -Module.unfreeze(*, recurse: bool = True, keys: str | List[str] | None = None, strict: bool = False) → Module#
                                            +Module.unfreeze(*, recurse: bool = True, keys: str | List[str] | None = None, strict: bool = False) → Module#

                                            Unfreeze the Module’s parameters or some of them.

                                            This function is idempotent ie unfreezing a model that is not frozen is a noop.

                                            @@ -946,13 +946,13 @@ a noop.

                                            Parameters:
                                              -

                                            • recurse (bool, optional) – If True then unfreeze the parameters of the +

                                            • recurse (bool, optional) – If True then unfreeze the parameters of the submodules as well. Default: True.

                                              • -

                                            • keys (str or list[str], optional) – If provided then only these +

                                            • keys (str or list[str], optional) – If provided then only these parameters will be unfrozen otherwise all the parameters of a module. For instance unfreeze all biases by calling module.unfreeze(keys="bias").

                                              • -

                                            • strict (bool, optional) – If set to True validate that the passed keys exist. +

                                            • strict (bool, optional) – If set to True validate that the passed keys exist. Default: False.

                                            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 19fac4f26..b0a22026c 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.29.2 documentation + mlx.nn.Module.update — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                            mlx.nn.Module.update#

                                            -Module.update(parameters: dict, strict: bool = True) → Module#
                                            +Module.update(parameters: dict, strict: bool = True) → Module#

                                            Replace the parameters of this Module with the provided ones in the dict of dicts and lists.

                                            Commonly used by the optimizer to change the model to the updated @@ -944,9 +944,9 @@ updated.

                                            Parameters:
                                              -

                                            • parameters (dict) – A complete or partial dictionary of the modules +

                                            • parameters (dict) – A complete or partial dictionary of the modules parameters.

                                              • -

                                            • strict (bool) – If True checks that parameters is a +

                                            • strict (bool) – If True checks that parameters is a subset of the module’s parameters. Default: True.

                                            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 62a0b2bf7..184efa346 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.29.2 documentation + mlx.nn.Module.update_modules — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                            mlx.nn.Module.update_modules#

                                            -Module.update_modules(modules: dict, strict: bool = True) → Module#
                                            +Module.update_modules(modules: dict, strict: bool = True) → Module#

                                            Replace the child modules of this Module instance with the provided ones in the dict of dicts and lists.

                                            It is the equivalent of Module.update() but for modules instead @@ -944,9 +944,9 @@ updated.

                                            Parameters:
                                              -

                                            • modules (dict) – A complete or partial dictionary of the module’s +

                                            • modules (dict) – A complete or partial dictionary of the module’s submodules.

                                              • -

                                            • strict (bool) – If True checks that modules is a +

                                            • strict (bool) – If True checks that modules is a subset of the child modules of this instance. Default: True.

                                            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 5239382f4..47a95ec13 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.29.2 documentation + mlx.nn.MultiHeadAttention — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                            mlx.nn.MultiHeadAttention#

                                            -class MultiHeadAttention(dims: int, num_heads: int, query_input_dims: int | None = None, key_input_dims: int | None = None, value_input_dims: int | None = None, value_dims: int | None = None, value_output_dims: int | None = None, bias: bool = False)#
                                            +class MultiHeadAttention(dims: int, num_heads: int, query_input_dims: int | None = None, key_input_dims: int | None = None, value_input_dims: int | None = None, value_dims: int | None = None, value_output_dims: int | None = None, bias: bool = False)#

                                            Implements the scaled dot product attention with multiple heads.

                                            Given inputs for queries, keys and values the MultiHeadAttention produces new values by aggregating information from the input values @@ -946,20 +946,20 @@ that should not be attended to.

                                            Parameters:
                                              -

                                            • dims (int) – The model dimensions. This is also the default +

                                            • dims (int) – The model dimensions. This is also the default value for the queries, keys, values, and the output.

                                              • -

                                            • num_heads (int) – The number of attention heads to use.

                                              • -

                                            • query_input_dims (int, optional) – The input dimensions of the queries. +

                                            • num_heads (int) – The number of attention heads to use.

                                              • +

                                            • query_input_dims (int, optional) – The input dimensions of the queries. Default: dims.

                                              • -

                                            • key_input_dims (int, optional) – The input dimensions of the keys. +

                                            • key_input_dims (int, optional) – The input dimensions of the keys. Default: dims.

                                              • -

                                            • value_input_dims (int, optional) – The input dimensions of the values. +

                                            • value_input_dims (int, optional) – The input dimensions of the values. Default: key_input_dims.

                                              • -

                                            • value_dims (int, optional) – The dimensions of the values after the +

                                            • value_dims (int, optional) – The dimensions of the values after the projection. Default: dims.

                                              • -

                                            • value_output_dims (int, optional) – The dimensions the new values will +

                                            • value_output_dims (int, optional) – The dimensions the new values will be projected to. Default: dims.

                                              • -

                                            • bias (bool, optional) – Whether or not to use a bias in the projections. +

                                            • bias (bool, optional) – Whether or not to use a bias in the projections. Default: False.

                                            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 39260ec0d..556750292 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.29.2 documentation + mlx.nn.PReLU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 60e50dd40..0ae38683c 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.29.2 documentation + mlx.nn.QuantizedEmbedding — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                            mlx.nn.QuantizedEmbedding#

                                            -class QuantizedEmbedding(num_embeddings: int, dims: int, group_size: int = 64, bits: int = 4, mode: str = 'affine')#
                                            +class QuantizedEmbedding(num_embeddings: int, dims: int, group_size: int = 64, bits: int = 4, mode: str = 'affine')#

                                            The same as Embedding but with a quantized weight matrix.

                                            QuantizedEmbedding also provides a from_embedding() classmethod to convert embedding layers to QuantizedEmbedding @@ -940,14 +940,14 @@ layers.

                                            Parameters:
                                              -

                                            • num_embeddings (int) – How many possible discrete tokens can we embed. +

                                            • num_embeddings (int) – How many possible discrete tokens can we embed. Usually called the vocabulary size.

                                              • -

                                            • dims (int) – The dimensionality of the embeddings.

                                              • -

                                            • group_size (int, optional) – The group size to use for the quantized +

                                            • dims (int) – The dimensionality of the embeddings.

                                              • +

                                            • group_size (int, optional) – The group size to use for the quantized weight. See quantize(). Default: 64.

                                              • -

                                            • bits (int, optional) – The bit width to use for the quantized weight. +

                                            • bits (int, optional) – The bit width to use for the quantized weight. See quantize(). Default: 4.

                                              • -

                                            • mode (str) – The quantization method to use (see +

                                            • mode (str) – The quantization method to use (see mlx.core.quantize()). Default: "affine".

                                            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 18bf8ddda..5835b3a85 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.29.2 documentation + mlx.nn.QuantizedLinear — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                            mlx.nn.QuantizedLinear#

                                            -class QuantizedLinear(input_dims: int, output_dims: int, bias: bool = True, group_size: int = 64, bits: int = 4, mode: str = 'affine')#
                                            +class QuantizedLinear(input_dims: int, output_dims: int, bias: bool = True, group_size: int = 64, bits: int = 4, mode: str = 'affine')#

                                            Applies an affine transformation to the input using a quantized weight matrix.

                                            It is the quantized equivalent of mlx.nn.Linear. For now its parameters are frozen and will not be included in any gradient computation @@ -942,15 +942,15 @@ convert linear layers to

                                            Parameters:
                                              -

                                            • input_dims (int) – The dimensionality of the input features.

                                              • -

                                            • output_dims (int) – The dimensionality of the output features.

                                              • -

                                            • bias (bool, optional) – If set to False then the layer will not use +

                                            • input_dims (int) – The dimensionality of the input features.

                                              • +

                                            • output_dims (int) – The dimensionality of the output features.

                                              • +

                                            • bias (bool, optional) – If set to False then the layer will not use a bias. Default: True.

                                              • -

                                            • group_size (int, optional) – The group size to use for the quantized +

                                            • group_size (int, optional) – The group size to use for the quantized weight. See quantize(). Default: 64.

                                              • -

                                            • bits (int, optional) – The bit width to use for the quantized weight. +

                                            • bits (int, optional) – The bit width to use for the quantized weight. See quantize(). Default: 4.

                                              • -

                                            • mode (str) – The quantization method to use (see +

                                            • mode (str) – The quantization method to use (see mlx.core.quantize()). Default: "affine".

                                            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 4eecdc7c9..2c2dafb71 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.29.2 documentation + mlx.nn.RMSNorm — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                            mlx.nn.RMSNorm#

                                            -class RMSNorm(dims: int, eps: float = 1e-05)#
                                            +class RMSNorm(dims: int, eps: float = 1e-05)#

                                            Applies Root Mean Square normalization [1] to the inputs.

                                            Computes

                                            @@ -945,8 +945,8 @@ document.write(`
                                            Parameters:
                                              -

                                            • dims (int) – The feature dimension of the input to normalize over

                                              • -

                                            • eps (float) – A small additive constant for numerical stability

                                              • +

                                            • dims (int) – The feature dimension of the input to normalize over

                                              • +

                                            • eps (float) – A small additive constant for numerical stability

                                            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 3fac47699..ead607053 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.29.2 documentation + mlx.nn.RNN — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                                            @@ -933,7 +933,7 @@ document.write(`

                                            mlx.nn.RNN#

                                            -class RNN(input_size: int, hidden_size: int, bias: bool = True, nonlinearity: Callable | None = None)#
                                            +class RNN(input_size: int, hidden_size: int, bias: bool = True, nonlinearity: Callable | None = None)#

                                            An Elman recurrent layer.

                                            The input is a sequence of shape NLD or LD where:

                                              @@ -951,9 +951,9 @@ time step, of shape
                                              Parameters:
                                                -

                                              • input_size (int) – Dimension of the input, D.

                                                • -

                                              • hidden_size (int) – Dimension of the hidden state, H.

                                                • -

                                              • bias (bool, optional) – Whether to use a bias. Default: True.

                                                • +

                                              • input_size (int) – Dimension of the input, D.

                                                • +

                                              • hidden_size (int) – Dimension of the hidden state, H.

                                                • +

                                              • bias (bool, optional) – Whether to use a bias. Default: True.

                                              • nonlinearity (callable, optional) – Non-linearity to use. If None, then func:tanh is used. Default: None.

                                              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 f54268326..e6cf9b79b 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.29.2 documentation + mlx.nn.ReLU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU2.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU2.html index e42c48cd3..2317cf124 100644 --- a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU2.html +++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU2.html @@ -8,7 +8,7 @@ - mlx.nn.ReLU2 — MLX 0.29.2 documentation + mlx.nn.ReLU2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 671bb8130..f25b98a76 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.29.2 documentation + mlx.nn.ReLU6 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 b0b9f34a2..484d347bc 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.29.2 documentation + mlx.nn.RoPE — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                              mlx.nn.RoPE#

                                              -class RoPE(dims: int, traditional: bool = False, base: float = 10000, scale: float = 1.0)#
                                              +class RoPE(dims: int, traditional: bool = False, base: float = 10000, scale: float = 1.0)#

                                              Implements the rotary positional encoding.

                                              The traditional implementation rotates consecutive pairs of elements in the feature dimension while the default implementation rotates pairs with @@ -942,13 +942,13 @@ Embedding.

                                              Parameters:
                                                -

                                              • dims (int) – The feature dimensions to be rotated. If the input feature +

                                              • dims (int) – The feature dimensions to be rotated. If the input feature is larger than dims then the rest is left unchanged.

                                                • -

                                              • traditional (bool, optional) – If set to True choose the traditional +

                                              • traditional (bool, optional) – If set to True choose the traditional implementation which is slightly less efficient. Default: False.

                                                • -

                                              • base (float, optional) – The base used to compute angular frequency for +

                                              • base (float, optional) – The base used to compute angular frequency for each dimension in the positional encodings. Default: 10000.

                                                • -

                                              • scale (float, optional) – The scale used to scale the positions. Default: 1.0.

                                                • +

                                              • scale (float, optional) – The scale used to scale the positions. Default: 1.0.

                                              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 e49871832..8a38fd40a 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.29.2 documentation + mlx.nn.SELU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 005c71388..a73673314 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.29.2 documentation + mlx.nn.Sequential — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -939,7 +939,7 @@ our functions have learnable parameters they should be implemented as nn.Module instances.

                                              Parameters:
                                              -

                                              modules (tuple of Callables) – The modules to call in order

                                              +

                                              modules (tuple of Callables) – The modules to call in order

                                              Methods

                                              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 cae2848e5..6f2c4e775 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.29.2 documentation + mlx.nn.SiLU — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 430f09b5d..a4fdb80ce 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.29.2 documentation + mlx.nn.Sigmoid — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 c02b04680..71ec5efd8 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.29.2 documentation + mlx.nn.SinusoidalPositionalEncoding — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,22 +933,22 @@ document.write(`

                                              mlx.nn.SinusoidalPositionalEncoding#

                                              -class SinusoidalPositionalEncoding(dims: int, min_freq: float = 0.0001, max_freq: float = 1, scale: float | None = None, cos_first: bool = False, full_turns: bool = False)#
                                              +class SinusoidalPositionalEncoding(dims: int, min_freq: float = 0.0001, max_freq: float = 1, scale: float | None = None, cos_first: bool = False, full_turns: bool = False)#

                                              Implements sinusoidal positional encoding.

                                              For more details see the paper Attention Is All You Need.

                                              Parameters:
                                                -

                                              • dims (int) – The dimensionality of the resulting positional embeddings.

                                                • -

                                              • min_freq (float, optional) – The minimum frequency expected. Default: +

                                              • dims (int) – The dimensionality of the resulting positional embeddings.

                                                • +

                                              • min_freq (float, optional) – The minimum frequency expected. Default: 0.0001.

                                                • -

                                              • max_freq (float, optional) – The maximum frequency expected. Default: +

                                              • max_freq (float, optional) – The maximum frequency expected. Default: 1.

                                                • -

                                              • scale (float, optional) – A multiplicative scale for the embeddings. +

                                              • scale (float, optional) – A multiplicative scale for the embeddings. Default: sqrt(2/dims).

                                                • -

                                              • cos_first (bool, optional) – If True embed using [cos(x); sin(x)] +

                                              • cos_first (bool, optional) – If True embed using [cos(x); sin(x)] instead of the reverse. Default: False.

                                                • -

                                              • full_turns (bool, optional) – If True multiply the frequencies with +

                                              • full_turns (bool, optional) – If True multiply the frequencies with \(2\pi\). Default: False.

                                              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 65bae602c..b117ff23b 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.29.2 documentation + mlx.nn.Softmax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 931ee9a81..2b46901b8 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.29.2 documentation + mlx.nn.Softmin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 fa9eddebb..d452201f7 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.29.2 documentation + mlx.nn.Softplus — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 8245158f0..775f12036 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.29.2 documentation + mlx.nn.Softshrink — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 d327248f9..4d4386f11 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.29.2 documentation + mlx.nn.Softsign — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 73d3e8700..3e44f9765 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.29.2 documentation + mlx.nn.Step — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                              mlx.nn.Step#

                                              -class Step(threshold: float = 0.0)#
                                              +class Step(threshold: float = 0.0)#

                                              Applies the Step Activation Function.

                                              This function implements a binary step activation, where the output is set to 1 if the input is greater than a specified threshold, and 0 otherwise.

                                              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 63adc372d..874c8f69c 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.29.2 documentation + mlx.nn.Tanh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 06389215b..56daabd9d 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.29.2 documentation + mlx.nn.Transformer — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -942,18 +942,18 @@ mechanism.

                                              Parameters:
                                                -

                                              • dims (int, optional) – The number of expected features in the +

                                              • dims (int, optional) – The number of expected features in the encoder/decoder inputs. Default: 512.

                                                • -

                                              • num_heads (int, optional) – The number of attention heads. Default: +

                                              • num_heads (int, optional) – The number of attention heads. Default: 8.

                                                • -

                                              • num_encoder_layers (int, optional) – The number of encoder layers in the +

                                              • num_encoder_layers (int, optional) – The number of encoder layers in the Transformer encoder. Default: 6.

                                                • -

                                              • num_decoder_layers (int, optional) – The number of decoder layers in the +

                                              • num_decoder_layers (int, optional) – The number of decoder layers in the Transformer decoder. Default: 6.

                                                • -

                                              • mlp_dims (int, optional) – The hidden dimension of the MLP block in each +

                                              • mlp_dims (int, optional) – The hidden dimension of the MLP block in each Transformer layer. Defaults to 4*dims if not provided. Default: None.

                                                • -

                                              • dropout (float, optional) – The dropout value for the Transformer +

                                              • dropout (float, optional) – The dropout value for the Transformer encoder and decoder. Dropout is used after each attention layer and the activation in the MLP layer. Default: 0.0.

                                              • activation (function, optional) – the activation function for the MLP @@ -962,10 +962,10 @@ hidden layer. Default: None.

                                              • custom_decoder (Module, optional) – A custom decoder to replace the standard Transformer decoder. Default: None.

                                                • -

                                              • norm_first (bool, optional) – if True, encoder and decoder layers +

                                              • norm_first (bool, optional) – if True, encoder and decoder layers will perform layer normalization before attention and MLP operations, otherwise after. Default: True.

                                                • -

                                              • checkpoint (bool, optional) – if True perform gradient checkpointing +

                                              • checkpoint (bool, optional) – if True perform gradient checkpointing to reduce the memory usage at the expense of more computation. Default: False.

                                              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 bd29a3649..b3df8ae34 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.29.2 documentation + mlx.nn.Upsample — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                              mlx.nn.Upsample#

                                              -class Upsample(scale_factor: float | Tuple, mode: Literal['nearest', 'linear', 'cubic'] = 'nearest', align_corners: bool = False)#
                                              +class Upsample(scale_factor: float | Tuple, mode: Literal['nearest', 'linear', 'cubic'] = 'nearest', align_corners: bool = False)#

                                              Upsample the input signal spatially.

                                              The spatial dimensions are by convention dimensions 1 to x.ndim - 2. The first is the batch dimension and the last is the feature @@ -954,13 +954,13 @@ output will be matching as will the bottom right edge.

                                              Parameters:
                                                -

                                              • scale_factor (float or tuple) – The multiplier for the spatial size. +

                                              • scale_factor (float or tuple) – The multiplier for the spatial size. If a float is provided, it is the multiplier for all spatial dimensions. Otherwise, the number of scale factors provided must match the number of spatial dimensions.

                                                • -

                                              • mode (str, optional) – The upsampling algorithm, either "nearest", +

                                              • mode (str, optional) – The upsampling algorithm, either "nearest", "linear" or "cubic". Default: "nearest".

                                                • -

                                              • align_corners (bool, optional) – Changes the way the corners are treated +

                                              • align_corners (bool, optional) – Changes the way the corners are treated during "linear" and "cubic" upsampling. See the note above and the examples below for more details. Default: False.

                                              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 36542f968..af797b9de 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.29.2 documentation + mlx.nn.init.constant — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,12 +932,12 @@ document.write(`

                                              mlx.nn.init.constant#

                                              -constant(value: float, dtype: Dtype = mlx.core.float32) → Callable[[array], array]#
                                              +constant(value: float, dtype: Dtype = mlx.core.float32) → Callable[[array], array]#

                                              An initializer that returns an array filled with value.

                                              Parameters:
                                                -

                                              • value (float) – The value to fill the array with.

                                                • +

                                              • value (float) – The value to fill the array with.

                                              • dtype (Dtype, optional) – The data type of the array. Default: float32.

                                              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 50327b8eb..656ab174e 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.29.2 documentation + mlx.nn.init.glorot_normal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                              mlx.nn.init.glorot_normal#

                                              -glorot_normal(dtype: Dtype = mlx.core.float32) → Callable[[array, float], array]#
                                              +glorot_normal(dtype: Dtype = mlx.core.float32) → Callable[[array, float], array]#

                                              A Glorot normal initializer.

                                              This initializer samples from a normal distribution with a standard deviation computed from the number of input (fan_in) and output @@ -952,7 +952,7 @@ with the same shape as the input, filled with samples from the Glorot normal distribution.

                                              Return type:
                                              -

                                              Callable[[array, float], array]

                                              +

                                              Callable[[array, float], array]

                                              Example

                                              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 42ca13169..8418973cf 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.29.2 documentation + mlx.nn.init.glorot_uniform — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                              mlx.nn.init.glorot_uniform#

                                              -glorot_uniform(dtype: Dtype = mlx.core.float32) → Callable[[array, float], array]#
                                              +glorot_uniform(dtype: Dtype = mlx.core.float32) → Callable[[array, float], array]#

                                              A Glorot uniform initializer.

                                              This initializer samples from a uniform distribution with a range computed from the number of input (fan_in) and output (fan_out) @@ -952,7 +952,7 @@ with the same shape as the input, filled with samples from the Glorot uniform distribution.

                                              Return type:
                                              -

                                              Callable[[array, float], array]

                                              +

                                              Callable[[array, float], array]

                                              Example

                                              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 05ffbdf1e..91a8a67f0 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.29.2 documentation + mlx.nn.init.he_normal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                              mlx.nn.init.he_normal#

                                              -he_normal(dtype: Dtype = mlx.core.float32) → Callable[[array, Literal['fan_in', 'fan_out'], float], array]#
                                              +he_normal(dtype: Dtype = mlx.core.float32) → Callable[[array, Literal['fan_in', 'fan_out'], float], array]#

                                              Build a He normal initializer.

                                              This initializer samples from a normal distribution with a standard deviation computed from the number of input (fan_in) or output @@ -955,7 +955,7 @@ array with the same shape as the input, filled with samples from the He normal distribution.

                                              Return type:
                                              -

                                              Callable[[array, str, float], array]

                                              +

                                              Callable[[array, str, float], array]

                                              Example

                                              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 a18daba28..e21752006 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.29.2 documentation + mlx.nn.init.he_uniform — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                              mlx.nn.init.he_uniform#

                                              -he_uniform(dtype: Dtype = mlx.core.float32) → Callable[[array, Literal['fan_in', 'fan_out'], float], array]#
                                              +he_uniform(dtype: Dtype = mlx.core.float32) → Callable[[array, Literal['fan_in', 'fan_out'], float], array]#

                                              A He uniform (Kaiming uniform) initializer.

                                              This initializer samples from a uniform distribution with a range computed from the number of input (fan_in) or output (fan_out) @@ -955,7 +955,7 @@ array with the same shape as the input, filled with samples from the He uniform distribution.

                                              Return type:
                                              -

                                              Callable[[array, str, float], array]

                                              +

                                              Callable[[array, str, float], array]

                                              Example

                                              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 927485943..fe80a2e6e 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.29.2 documentation + mlx.nn.init.identity — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                              mlx.nn.init.identity#

                                              -identity(dtype: Dtype = mlx.core.float32) → Callable[[array], array]#
                                              +identity(dtype: Dtype = mlx.core.float32) → Callable[[array], array]#

                                              An initializer that returns an identity matrix.

                                              Parameters:
                                              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 792882959..ce3c77391 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.29.2 documentation + mlx.nn.init.normal — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                                              mlx.nn.init.normal#

                                              -normal(mean: float = 0.0, std: float = 1.0, dtype: Dtype = mlx.core.float32) → Callable[[array], array]#
                                              +normal(mean: float = 0.0, std: float = 1.0, dtype: Dtype = mlx.core.float32) → Callable[[array], array]#

                                              An initializer that returns samples from a normal distribution.

                                              Parameters:
                                                -

                                              • mean (float, optional) – Mean of the normal distribution. Default: +

                                              • mean (float, optional) – Mean of the normal distribution. Default: 0.0.

                                                • -

                                              • std (float, optional) – Standard deviation of the normal distribution. +

                                              • std (float, optional) – Standard deviation of the normal distribution. Default: 1.0.

                                              • dtype (Dtype, optional) – The data type of the array. Default: float32.

                                                • 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 d745c87f8..978038633 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.29.2 documentation + mlx.nn.init.uniform — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                                                mlx.nn.init.uniform#

                                                -uniform(low: float = 0.0, high: float = 1.0, dtype: Dtype = mlx.core.float32) → Callable[[array], array]#
                                                +uniform(low: float = 0.0, high: float = 1.0, dtype: Dtype = mlx.core.float32) → Callable[[array], array]#

                                                An initializer that returns samples from a uniform distribution.

                                                Parameters:
                                                  -

                                                • low (float, optional) – The lower bound of the uniform distribution. +

                                                • low (float, optional) – The lower bound of the uniform distribution. Default: 0.0.

                                                  • -

                                                • high (float, optional) – The upper bound of the uniform distribution. +

                                                • high (float, optional) – The upper bound of the uniform distribution. Default: 1.0

                                                • dtype (Dtype, optional) – The data type of the array. Default: float32.

                                                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 00f05e468..417b975d3 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.29.2 documentation + mlx.nn.celu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 dc547ff58..91bb28a29 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.29.2 documentation + mlx.nn.elu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 4daee8b73..b3ee46c4d 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.29.2 documentation + mlx.nn.gelu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 d2f799ad6..6e8fb74f1 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.29.2 documentation + mlx.nn.gelu_approx — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 3a4599a22..3d1b218ca 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.29.2 documentation + mlx.nn.gelu_fast_approx — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 f5db8a539..1e7157809 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.29.2 documentation + mlx.nn.glu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                                mlx.nn.glu#

                                                -class glu(x: array, axis: int = -1)#
                                                +class glu(x: array, axis: int = -1)#

                                                Applies the gated linear unit function.

                                                This function splits the axis dimension of the input into two halves (\(a\) and \(b\)) and applies \(a * \sigma(b)\).

                                                @@ -941,7 +941,7 @@ document.write(` \[\textrm{GLU}(x) = a * \sigma(b)\]
                                                Parameters:
                                                -

                                                axis (int) – The dimension to split along. Default: -1

                                                +

                                                axis (int) – The dimension to split along. Default: -1

                                                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 a944e1836..a22333100 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.29.2 documentation + mlx.nn.hard_shrink — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 e03cb0dba..080e287f1 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.29.2 documentation + mlx.nn.hard_tanh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 f1ffadbab..acc74df2f 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.29.2 documentation + mlx.nn.hardswish — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 664e80126..616f9467f 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.29.2 documentation + mlx.nn.leaky_relu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 7cbf8275b..fbed77924 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.29.2 documentation + mlx.nn.log_sigmoid — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 0e5122f27..da6da5c21 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.29.2 documentation + mlx.nn.log_softmax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 ceef7b898..9e4304b6b 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.29.2 documentation + mlx.nn.losses.binary_cross_entropy — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                                mlx.nn.losses.binary_cross_entropy#

                                                -class binary_cross_entropy(inputs: array, targets: array, weights: array | None = None, with_logits: bool = True, reduction: Literal['none', 'mean', 'sum'] = 'mean')#
                                                +class binary_cross_entropy(inputs: array, targets: array, weights: array | None = None, with_logits: bool = True, reduction: Literal['none', 'mean', 'sum'] = 'mean')#

                                                Computes the binary cross entropy loss.

                                                By default, this function takes the pre-sigmoid logits, which results in a faster and more precise loss. For improved numerical stability when with_logits=False, @@ -944,9 +944,9 @@ of -100

                                                inputs (array) – The predicted values. If with_logits is True, then inputs are unnormalized logits. Otherwise, inputs are probabilities.

                                              • targets (array) – The binary target values in {0, 1}.

                                                • -

                                              • with_logits (bool, optional) – Whether inputs are logits. Default: True.

                                                • +

                                              • with_logits (bool, optional) – Whether inputs are logits. Default: True.

                                              • weights (array, optional) – Optional weights for each target. Default: None.

                                                • -

                                              • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                              • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'mean'.

                                              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 9558b54c4..5810fd3b1 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.29.2 documentation + mlx.nn.losses.cosine_similarity_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                              mlx.nn.losses.cosine_similarity_loss#

                                              -class cosine_similarity_loss(x1: array, x2: array, axis: int = 1, eps: float = 1e-08, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                              +class cosine_similarity_loss(x1: array, x2: array, axis: int = 1, eps: float = 1e-08, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                              Computes the cosine similarity between the two inputs.

                                              The cosine similarity loss is given by

                                              @@ -943,10 +943,10 @@ document.write(`

                                              • x1 (mx.array) – The first set of inputs.

                                              • x2 (mx.array) – The second set of inputs.

                                                • -

                                              • axis (int, optional) – The embedding axis. Default: 1.

                                                • -

                                              • eps (float, optional) – The minimum value of the denominator used for +

                                              • axis (int, optional) – The embedding axis. Default: 1.

                                                • +

                                              • eps (float, optional) – The minimum value of the denominator used for numerical stability. Default: 1e-8.

                                                • -

                                              • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                              • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                              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 c6e255820..0ced8b992 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.29.2 documentation + mlx.nn.losses.cross_entropy — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                              mlx.nn.losses.cross_entropy#

                                              -class cross_entropy(logits: array, targets: array, weights: array | None = None, axis: int = -1, label_smoothing: float = 0.0, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                              +class cross_entropy(logits: array, targets: array, weights: array | None = None, axis: int = -1, label_smoothing: float = 0.0, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                              Computes the cross entropy loss.

                                              Parameters:
                                              @@ -945,9 +945,9 @@ the axistargets shape should be the same as the logits shape.

                                            • weights (array, optional) – Optional weights for each target. Default: None.

                                              • -

                                            • axis (int, optional) – The axis over which to compute softmax. Default: -1.

                                              • -

                                            • label_smoothing (float, optional) – Label smoothing factor. Default: 0.

                                              • -

                                            • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                            • axis (int, optional) – The axis over which to compute softmax. Default: -1.

                                              • +

                                            • label_smoothing (float, optional) – Label smoothing factor. Default: 0.

                                              • +

                                            • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                            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 bd842d972..2af90c210 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.29.2 documentation + mlx.nn.losses.gaussian_nll_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                            mlx.nn.losses.gaussian_nll_loss#

                                            -class gaussian_nll_loss(inputs: array, targets: array, vars: array, full: bool = False, eps: float = 1e-06, reduction: Literal['none', 'mean', 'sum'] = 'mean')#
                                            +class gaussian_nll_loss(inputs: array, targets: array, vars: array, full: bool = False, eps: float = 1e-06, reduction: Literal['none', 'mean', 'sum'] = 'mean')#

                                            Computes the negative log likelihood loss for a Gaussian distribution.

                                            The loss is given by:

                                            @@ -948,11 +948,11 @@ predicted variances.

                                          • inputs (array) – The predicted expectation of the Gaussian distribution.

                                          • targets (array) – The target values (samples from the Gaussian distribution).

                                          • vars (array) – The predicted variance of the Gaussian distribution.

                                            • -

                                          • full (bool, optional) – Whether to include the constant term in the loss calculation. +

                                          • full (bool, optional) – Whether to include the constant term in the loss calculation. Default: False.

                                            • -

                                          • eps (float, optional) – Small positive constant for numerical stability. +

                                          • eps (float, optional) – Small positive constant for numerical stability. Default: 1e-6.

                                            • -

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                          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 648441007..3cacc8c3a 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.29.2 documentation + mlx.nn.losses.hinge_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                          mlx.nn.losses.hinge_loss#

                                          -class hinge_loss(inputs: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                          +class hinge_loss(inputs: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                          Computes the hinge loss between inputs and targets.

                                          \[\text{hinge}(y, y_{\text{pred}}) = \max(0, 1 - y \cdot y_{\text{pred}})\]
                                          @@ -942,7 +942,7 @@ document.write(`

                                          • inputs (array) – The predicted values.

                                          • targets (array) – The target values. They should be -1 or 1.

                                            • -

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                          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 9cd48d4fe..17b3b8aa9 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.29.2 documentation + mlx.nn.losses.huber_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                          mlx.nn.losses.huber_loss#

                                          -class huber_loss(inputs: array, targets: array, delta: float = 1.0, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                          +class huber_loss(inputs: array, targets: array, delta: float = 1.0, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                          Computes the Huber loss between inputs and targets.

                                          \[\begin{split}l_{\delta}(a) = @@ -946,9 +946,9 @@ document.write(`

                                          • inputs (array) – The predicted values.

                                          • targets (array) – The target values.

                                            • -

                                          • delta (float, optional) – The threshold at which to change between L1 and L2 loss. +

                                          • delta (float, optional) – The threshold at which to change between L1 and L2 loss. Default: 1.0.

                                            • -

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                          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 b52418263..d8f9268ae 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.29.2 documentation + mlx.nn.losses.kl_div_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                          mlx.nn.losses.kl_div_loss#

                                          -class kl_div_loss(inputs: array, targets: array, axis: int = -1, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                          +class kl_div_loss(inputs: array, targets: array, axis: int = -1, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                          Computes the Kullback-Leibler divergence loss.

                                          Computes the following when reduction == 'none':

                                          mx.exp(targets) * (targets - inputs).sum(axis)
@@ -943,8 +943,8 @@ document.write(`
 
                                            
 
                                          • inputs (array) – Log probabilities for the predicted distribution.
                                            • 
 
                                          • targets (array) – Log probabilities for the target distribution.
                                            • 
-
                                          • axis (int, optional) – The distribution axis. Default: -1.
                                            • 
-
                                          • reduction (str, optional) – Specifies the reduction to apply to the output:
+
                                          • axis (int, optional) – The distribution axis. Default: -1.
                                            • 
+
                                          • reduction (str, optional) – Specifies the reduction to apply to the output:
 'none' | 'mean' | 'sum'. Default: 'none'.
                                            • 
 
                                          
 
                                          
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 5054cb767..296f005ce 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.29.2 documentation
+    mlx.nn.losses.l1_loss — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                                          @@ -932,14 +932,14 @@ document.write(`

                                          mlx.nn.losses.l1_loss#

                                          -class l1_loss(predictions: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'mean')#
                                          +class l1_loss(predictions: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'mean')#

                                          Computes the L1 loss.

                                          Parameters:

                                          • predictions (array) – The predicted values.

                                          • targets (array) – The target values.

                                            • -

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'mean'.

                                          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 741a83811..0843fc389 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.29.2 documentation + mlx.nn.losses.log_cosh_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                                          @@ -933,7 +933,7 @@ document.write(`

                                          mlx.nn.losses.log_cosh_loss#

                                          -class log_cosh_loss(inputs: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                          +class log_cosh_loss(inputs: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                          Computes the log cosh loss between inputs and targets.

                                          Logcosh acts like L2 loss for small errors, ensuring stable gradients, and like the L1 loss for large errors, reducing sensitivity to outliers. This @@ -947,7 +947,7 @@ dual behavior offers a balanced, robust approach for regression tasks.

                                          • inputs (array) – The predicted values.

                                          • targets (array) – The target values.

                                            • -

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                          • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                          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 20a00a807..eef32591a 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.29.2 documentation + mlx.nn.losses.margin_ranking_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                          mlx.nn.losses.margin_ranking_loss#

                                          -class margin_ranking_loss(inputs1: array, inputs2: array, targets: array, margin: float = 0.0, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                          +class margin_ranking_loss(inputs1: array, inputs2: array, targets: array, margin: float = 0.0, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                          Calculate the margin ranking loss that loss given inputs \(x_1\), \(x_2\) and a label \(y\) (containing 1 or -1).

                                          The loss is given by:

                                          @@ -948,9 +948,9 @@ represents inputs2<

                                        • inputs2 (array) – Scores for the second input.

                                        • targets (array) – Labels indicating whether samples in inputs1 should be ranked higher than samples in inputs2. Values should be 1 or -1.

                                          • -

                                        • margin (float, optional) – The margin by which the scores should be separated. +

                                        • margin (float, optional) – The margin by which the scores should be separated. Default: 0.0.

                                          • -

                                        • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                        • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                        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 d0af596ac..ce09e238a 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.29.2 documentation + mlx.nn.losses.mse_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                                        mlx.nn.losses.mse_loss#

                                        -class mse_loss(predictions: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'mean')#
                                        +class mse_loss(predictions: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'mean')#

                                        Computes the mean squared error loss.

                                        Parameters:

                                        • predictions (array) – The predicted values.

                                        • targets (array) – The target values.

                                          • -

                                        • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                        • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'mean'.

                                        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 764315997..e627365d1 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.29.2 documentation + mlx.nn.losses.nll_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

                                        mlx.nn.losses.nll_loss#

                                        -class nll_loss(inputs: array, targets: array, axis: int = -1, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                        +class nll_loss(inputs: array, targets: array, axis: int = -1, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                        Computes the negative log likelihood loss.

                                        Parameters:

                                        • inputs (array) – The predicted distribution in log space.

                                        • targets (array) – The target values.

                                          • -

                                        • axis (int, optional) – The distribution axis. Default: -1.

                                          • -

                                        • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                        • axis (int, optional) – The distribution axis. Default: -1.

                                          • +

                                        • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                        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 82eb06423..023643298 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.29.2 documentation + mlx.nn.losses.smooth_l1_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                        mlx.nn.losses.smooth_l1_loss#

                                        -class smooth_l1_loss(predictions: array, targets: array, beta: float = 1.0, reduction: Literal['none', 'mean', 'sum'] = 'mean')#
                                        +class smooth_l1_loss(predictions: array, targets: array, beta: float = 1.0, reduction: Literal['none', 'mean', 'sum'] = 'mean')#

                                        Computes the smooth L1 loss.

                                        The smooth L1 loss is a variant of the L1 loss which replaces the absolute difference with a squared difference when the absolute difference is less @@ -949,9 +949,9 @@ than beta

                                        • predictions (array) – Predicted values.

                                        • targets (array) – Ground truth values.

                                          • -

                                        • beta (float, optional) – The threshold after which the loss changes +

                                        • beta (float, optional) – The threshold after which the loss changes from the squared to the absolute difference. Default: 1.0.

                                          • -

                                        • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                        • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'mean'.

                                        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 19a337387..75d73b5e8 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.29.2 documentation + mlx.nn.losses.triplet_loss — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                        mlx.nn.losses.triplet_loss#

                                        -class triplet_loss(anchors: array, positives: array, negatives: array, axis: int = -1, p: int = 2, margin: float = 1.0, eps: float = 1e-06, reduction: Literal['none', 'mean', 'sum'] = 'none')#
                                        +class triplet_loss(anchors: array, positives: array, negatives: array, axis: int = -1, p: int = 2, margin: float = 1.0, eps: float = 1e-06, reduction: Literal['none', 'mean', 'sum'] = 'none')#

                                        Computes the triplet loss for a set of anchor, positive, and negative samples. Margin is represented with alpha in the math section.

                                        @@ -944,11 +944,11 @@ Margin is represented with alpha in the math section.

                                      • anchors (array) – The anchor samples.

                                      • positives (array) – The positive samples.

                                      • negatives (array) – The negative samples.

                                        • -

                                      • axis (int, optional) – The distribution axis. Default: -1.

                                        • -

                                      • p (int, optional) – The norm degree for pairwise distance. Default: 2.

                                        • -

                                      • margin (float, optional) – Margin for the triplet loss. Defaults to 1.0.

                                        • -

                                      • eps (float, optional) – Small positive constant to prevent numerical instability. Defaults to 1e-6.

                                        • -

                                      • reduction (str, optional) – Specifies the reduction to apply to the output: +

                                      • axis (int, optional) – The distribution axis. Default: -1.

                                        • +

                                      • p (int, optional) – The norm degree for pairwise distance. Default: 2.

                                        • +

                                      • margin (float, optional) – Margin for the triplet loss. Defaults to 1.0.

                                        • +

                                      • eps (float, optional) – Small positive constant to prevent numerical instability. Defaults to 1e-6.

                                        • +

                                      • reduction (str, optional) – Specifies the reduction to apply to the output: 'none' | 'mean' | 'sum'. Default: 'none'.

                                      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 acf93328b..1180375a1 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.29.2 documentation + mlx.nn.mish — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 15baf21dd..60e0a9835 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.29.2 documentation + mlx.nn.prelu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 311b51fbd..dad69ca83 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.29.2 documentation + mlx.nn.relu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu2.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu2.html index 2f723dcda..0d1ca5236 100644 --- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu2.html +++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu2.html @@ -8,7 +8,7 @@ - mlx.nn.relu2 — MLX 0.29.2 documentation + mlx.nn.relu2 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 70aa9b5ed..201f70c11 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.29.2 documentation + mlx.nn.relu6 — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 672186102..fd9219412 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.29.2 documentation + mlx.nn.selu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 58776216a..8dd437c53 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.29.2 documentation + mlx.nn.sigmoid — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 662f3bbf6..b09539eef 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.29.2 documentation + mlx.nn.silu — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 5b0ea1649..473fd0e7a 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.29.2 documentation + mlx.nn.softmax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 05ceef2d0..57ef53ddc 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.29.2 documentation + mlx.nn.softmin — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 b2ee390ab..4ea9dfb0f 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.29.2 documentation + mlx.nn.softplus — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 8850eb469..aefd97e87 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.29.2 documentation + mlx.nn.softshrink — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.nn.softshrink#

                                      -class softshrink(x, lambd: float = 0.5)#
                                      +class softshrink(x, lambd: float = 0.5)#

                                      Applies the Softshrink activation function.

                                      \[\begin{split}\text{softshrink}(x) = \begin{cases} 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 4a1127bdc..cd675c937 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.29.2 documentation + mlx.nn.step — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                                      @@ -933,7 +933,7 @@ document.write(`

                                      mlx.nn.step#

                                      -class step(x: array, threshold: float = 0.0)#
                                      +class step(x: array, threshold: float = 0.0)#

                                      Applies the Step Activation Function.

                                      This function implements a binary step activation, where the output is set to 1 if the input is greater than a specified threshold, and 0 otherwise.

                                      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 8a1dead9a..e73eb1360 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.29.2 documentation + mlx.nn.tanh — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/nn/functions.html b/docs/build/html/python/nn/functions.html index 9100b1cf4..c968bb4c2 100644 --- a/docs/build/html/python/nn/functions.html +++ b/docs/build/html/python/nn/functions.html @@ -8,7 +8,7 @@ - Functions — MLX 0.29.2 documentation + Functions — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/nn/init.html b/docs/build/html/python/nn/init.html index 8008cf2db..856f22601 100644 --- a/docs/build/html/python/nn/init.html +++ b/docs/build/html/python/nn/init.html @@ -8,7 +8,7 @@ - Initializers — MLX 0.29.2 documentation + Initializers — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/nn/layers.html b/docs/build/html/python/nn/layers.html index a5e030512..b61765a92 100644 --- a/docs/build/html/python/nn/layers.html +++ b/docs/build/html/python/nn/layers.html @@ -8,7 +8,7 @@ - Layers — MLX 0.29.2 documentation + Layers — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/nn/losses.html b/docs/build/html/python/nn/losses.html index a668e7655..765c53b4d 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.29.2 documentation + Loss Functions — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/nn/module.html b/docs/build/html/python/nn/module.html index 654d2a400..7aea21eb6 100644 --- a/docs/build/html/python/nn/module.html +++ b/docs/build/html/python/nn/module.html @@ -8,7 +8,7 @@ - Module — MLX 0.29.2 documentation + Module — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/ops.html b/docs/build/html/python/ops.html index 2e263c31e..bcd820d4a 100644 --- a/docs/build/html/python/ops.html +++ b/docs/build/html/python/ops.html @@ -8,7 +8,7 @@ - Operations — MLX 0.29.2 documentation + Operations — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/optimizers.html b/docs/build/html/python/optimizers.html index ec08f8020..8f93dc39a 100644 --- a/docs/build/html/python/optimizers.html +++ b/docs/build/html/python/optimizers.html @@ -8,7 +8,7 @@ - Optimizers — MLX 0.29.2 documentation + Optimizers — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 1df297d48..72239965a 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.29.2 documentation + mlx.optimizers.AdaDelta — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.optimizers.AdaDelta#

                                      -class AdaDelta(learning_rate: float | Callable[[array], array], rho: float = 0.9, eps: float = 1e-06)#
                                      +class AdaDelta(learning_rate: float | Callable[[array], array], rho: float = 0.9, eps: float = 1e-06)#

                                      The AdaDelta optimizer with a learning rate [1].

                                      Our AdaDelta implementation follows the original paper. In detail,

                                      [1]: Zeiler, M.D., 2012. ADADELTA: an adaptive learning rate method. arXiv preprint arXiv:1212.5701.

                                      @@ -945,10 +945,10 @@ w_{t+1} &= w_t - \lambda \Delta w_{t+1}\end{split}\]
                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • -

                                      • rho (float, optional) – The coefficient \(\rho\) used for computing a +

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • +

                                      • rho (float, optional) – The coefficient \(\rho\) used for computing a running average of squared gradients. Default: 0.9

                                        • -

                                      • eps (float, optional) – The term \(\epsilon\) added to the denominator to improve +

                                      • eps (float, optional) – The term \(\epsilon\) added to the denominator to improve numerical stability. Default: 1e-8

                                      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 a4869c6b1..f6545d8f1 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.29.2 documentation + mlx.optimizers.Adafactor — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,35 +933,35 @@ document.write(`

                                      mlx.optimizers.Adafactor#

                                      -class Adafactor(learning_rate: float | Callable[[array], array] | None = None, eps: Tuple[float, float] = (1e-30, 0.001), clip_threshold: float = 1.0, decay_rate: float = -0.8, beta_1: float | None = None, weight_decay: float = 0.0, scale_parameter: bool = True, relative_step: bool = True, warmup_init: bool = False)#
                                      +class Adafactor(learning_rate: float | Callable[[array], array] | None = None, eps: Tuple[float, float] = (1e-30, 0.001), clip_threshold: float = 1.0, decay_rate: float = -0.8, beta_1: float | None = None, weight_decay: float = 0.0, scale_parameter: bool = True, relative_step: bool = True, warmup_init: bool = False)#

                                      The Adafactor optimizer.

                                      Our Adafactor implementation follows the original paper: Adafactor: Adaptive Learning Rates with Sublinear Memory Cost

                                      Parameters:
                                        -

                                      • learning_rate (float or callable, optional) – The learning rate. +

                                      • learning_rate (float or callable, optional) – The learning rate. Default: None.

                                        • -

                                      • eps (tuple(float, float), optional) – The first term \(\epsilon_1\) +

                                      • eps (tuple(float, float), optional) – The first term \(\epsilon_1\) added to the square of the gradients to improve numerical stability and the second term \(\epsilon_2\) is used for parameter scaling if parameter_scale is set to True. Default: (1e-30, 1e-3).

                                        • -

                                      • clip_threshold (float, optional) – Clips the unscaled update at +

                                      • clip_threshold (float, optional) – Clips the unscaled update at clip_threshold. Default: 1.0.

                                        • -

                                      • decay_rate (float, optional) – Coefficient for the running average +

                                      • decay_rate (float, optional) – Coefficient for the running average of the squared gradient. Default: -0.8.

                                        • -

                                      • beta_1 (float, optional) – If set to a value bigger than zero +

                                      • beta_1 (float, optional) – If set to a value bigger than zero then first moment will be used. Default: None.

                                        • -

                                      • weight_decay (float, optional) – The weight decay \(\lambda\). +

                                      • weight_decay (float, optional) – The weight decay \(\lambda\). Default: 0.0.

                                        • -

                                      • scale_parameter (bool, optional) – If set to True the learning rate +

                                      • scale_parameter (bool, optional) – If set to True the learning rate will be scaled by \(\max(\epsilon_1, \text{RMS}(w_{t-1}))\). Default: True.

                                        • -

                                      • relative_step (bool, optional) – If set to True the learning_rate +

                                      • relative_step (bool, optional) – If set to True the learning_rate will be ignored and relative step size will be computed. Default: True.

                                        • -

                                      • warmup_init (bool, optional) – If set to True then the relative +

                                      • warmup_init (bool, optional) – If set to True then the relative step size will be calculated by the current step. Default: False.

                                      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 6390744c5..b98340680 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.29.2 documentation + mlx.optimizers.Adagrad — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.optimizers.Adagrad#

                                      -class Adagrad(learning_rate: float | Callable[[array], array], eps: float = 1e-08)#
                                      +class Adagrad(learning_rate: float | Callable[[array], array], eps: float = 1e-08)#

                                      The Adagrad optimizer [1].

                                      Our Adagrad implementation follows the original paper. In detail,

                                      [1]: Duchi, J., Hazan, E. and Singer, Y., 2011. Adaptive subgradient methods @@ -944,8 +944,8 @@ w_{t+1} &= w_t - \lambda \frac{g_t}{\sqrt{v_{t+1}} + \epsilon}\end{split}\]<

                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • -

                                      • eps (float, optional) – The term \(\epsilon\) added to the +

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • +

                                      • eps (float, optional) – The term \(\epsilon\) added to the denominator to improve numerical stability. Default: 1e-8

                                      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 68c46e99f..3d0f034c6 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.29.2 documentation + mlx.optimizers.Adam — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.optimizers.Adam#

                                      -class Adam(learning_rate: float | Callable[[array], array], betas: List[float] = [0.9, 0.999], eps: float = 1e-08, bias_correction: bool = False)#
                                      +class Adam(learning_rate: float | Callable[[array], array], betas: List[float] = [0.9, 0.999], eps: float = 1e-08, bias_correction: bool = False)#

                                      The Adam optimizer [1]. In detail,

                                      [1]: Kingma, D.P. and Ba, J., 2015. Adam: A method for stochastic optimization. ICLR 2015.

                                      @@ -944,13 +944,13 @@ w_{t+1} &= w_t - \lambda \frac{m_{t+1}}{\sqrt{v_{t+1}} + \epsilon}\end{split
                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • -

                                      • betas (Tuple[float, float], optional) – The coefficients +

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • +

                                      • betas (Tuple[float, float], optional) – The coefficients \((\beta_1, \beta_2)\) used for computing running averages of the gradient and its square. Default: (0.9, 0.999)

                                        • -

                                      • eps (float, optional) – The term \(\epsilon\) added to the +

                                      • eps (float, optional) – The term \(\epsilon\) added to the denominator to improve numerical stability. Default: 1e-8

                                        • -

                                      • bias_correction (bool, optional) – If set to True, bias correction +

                                      • bias_correction (bool, optional) – If set to True, bias correction is applied. Default: False

                                      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 ebdfcc124..6f69b6269 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.29.2 documentation + mlx.optimizers.AdamW — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.optimizers.AdamW#

                                      -class AdamW(learning_rate: float | Callable[[array], array], betas: List[float] = [0.9, 0.999], eps: float = 1e-08, weight_decay: float = 0.01, bias_correction: bool = False)#
                                      +class AdamW(learning_rate: float | Callable[[array], array], betas: List[float] = [0.9, 0.999], eps: float = 1e-08, weight_decay: float = 0.01, bias_correction: bool = False)#

                                      The AdamW optimizer [1]. We update the weights with a weight_decay (\(\lambda\)) value:

                                      [1]: Loshchilov, I. and Hutter, F., 2019. Decoupled weight decay @@ -945,15 +945,15 @@ w_{t+1} &= w_t - \alpha (\frac{m_{t+1}}{\sqrt{v_{t+1}} + \epsilon} + \lambda

                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate \(\alpha\).

                                        • -

                                      • betas (Tuple[float, float], optional) – The coefficients +

                                      • learning_rate (float or callable) – The learning rate \(\alpha\).

                                        • +

                                      • betas (Tuple[float, float], optional) – The coefficients \((\beta_1, \beta_2)\) used for computing running averages of the gradient and its square. Default: (0.9, 0.999)

                                        • -

                                      • eps (float, optional) – The term \(\epsilon\) added to the +

                                      • eps (float, optional) – The term \(\epsilon\) added to the denominator to improve numerical stability. Default: 1e-8

                                        • -

                                      • weight_decay (float, optional) – The weight decay \(\lambda\). +

                                      • weight_decay (float, optional) – The weight decay \(\lambda\). Default: 0.01.

                                        • -

                                      • bias_correction (bool, optional) – If set to True, bias correction +

                                      • bias_correction (bool, optional) – If set to True, bias correction is applied. Default: False

                                      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 ae80627dc..824a43890 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.29.2 documentation + mlx.optimizers.Adamax — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.optimizers.Adamax#

                                      -class Adamax(learning_rate: float | Callable[[array], array], betas: List[float] = [0.9, 0.999], eps: float = 1e-08)#
                                      +class Adamax(learning_rate: float | Callable[[array], array], betas: List[float] = [0.9, 0.999], eps: float = 1e-08)#

                                      The Adamax optimizer, a variant of Adam based on the infinity norm [1].

                                      Our Adam implementation follows the original paper and omits the bias correction in the first and second moment estimates. In detail,

                                      @@ -946,11 +946,11 @@ w_{t+1} &= w_t - \lambda \frac{m_{t+1}}{v_{t+1} + \epsilon}\end{split}\]
                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • -

                                      • betas (Tuple[float, float], optional) – The coefficients +

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • +

                                      • betas (Tuple[float, float], optional) – The coefficients \((\beta_1, \beta_2)\) used for computing running averages of the gradient and its square. Default: (0.9, 0.999)

                                        • -

                                      • eps (float, optional) – The term \(\epsilon\) added to the +

                                      • eps (float, optional) – The term \(\epsilon\) added to the denominator to improve numerical stability. Default: 1e-8

                                      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 2741fe6ca..b86ea753d 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.29.2 documentation + mlx.optimizers.Lion — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.optimizers.Lion#

                                      -class Lion(learning_rate: float | Callable[[array], array], betas: List[float] = [0.9, 0.99], weight_decay: float = 0.0)#
                                      +class Lion(learning_rate: float | Callable[[array], array], betas: List[float] = [0.9, 0.99], weight_decay: float = 0.0)#

                                      The Lion optimizer [1].

                                      Since updates are computed through the sign operation, they tend to have larger norm than for other optimizers such as SGD and Adam. @@ -950,11 +950,11 @@ w_{t + 1} &= w_t - \eta (\text{sign}(c_t) + \lambda w_t)\end{split}\]

                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate \(\eta\).

                                        • -

                                      • betas (Tuple[float, float], optional) – The coefficients +

                                      • learning_rate (float or callable) – The learning rate \(\eta\).

                                        • +

                                      • betas (Tuple[float, float], optional) – The coefficients \((\beta_1, \beta_2)\) used for computing the gradient momentum and update direction. Default: (0.9, 0.99)

                                        • -

                                      • weight_decay (float, optional) – The weight decay \(\lambda\). Default: 0.0

                                        • +

                                      • weight_decay (float, optional) – The weight decay \(\lambda\). Default: 0.0

                                      diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.MultiOptimizer.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.MultiOptimizer.html index a825eadf6..5f12c4340 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.MultiOptimizer.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.MultiOptimizer.html @@ -8,7 +8,7 @@ - mlx.optimizers.MultiOptimizer — MLX 0.29.2 documentation + mlx.optimizers.MultiOptimizer — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                      mlx.optimizers.MultiOptimizer#

                                      -class MultiOptimizer(optimizers, filters: list = [])#
                                      +class MultiOptimizer(optimizers, filters: list = [])#

                                      Wraps a list of optimizers with corresponding weight predicates/filters to make it easy to use different optimizers for different weights.

                                      The predicates take the full “path” of the weight and the weight itself and @@ -942,8 +942,8 @@ given for it.

                                      Parameters:
                                        -

                                      • optimizers (list[Optimizer]) – A list of optimizers to delegate to

                                        • -

                                      • filters (list[Callable[[str, array], bool]) – A list of predicates that +

                                      • optimizers (list[Optimizer]) – A list of optimizers to delegate to

                                        • +

                                      • filters (list[Callable[[str, array], bool]) – A list of predicates that should be one less than the provided optimizers.

                                      diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Muon.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Muon.html index dea0ad045..886303ac7 100644 --- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Muon.html +++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Muon.html @@ -8,7 +8,7 @@ - mlx.optimizers.Muon — MLX 0.29.2 documentation + mlx.optimizers.Muon — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                      mlx.optimizers.Muon#

                                      -class Muon(learning_rate: float | Callable[[array], array], momentum: float = 0.95, weight_decay: float = 0.01, nesterov: bool = True, ns_steps: int = 5)#
                                      +class Muon(learning_rate: float | Callable[[array], array], momentum: float = 0.95, weight_decay: float = 0.01, nesterov: bool = True, ns_steps: int = 5)#

                                      The Muon optimizer.

                                      Our Muon (MomentUm Orthogonalized by Newton-schulz) optimizer follows the original implementation: Muon: An optimizer for hidden layers in neural @@ -950,13 +950,13 @@ dimensions.

                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate.

                                        • -

                                      • momentum (float, optional) – The momentum strength. Default: 0.95

                                        • -

                                      • weight_decay (float, optional) – The weight decay (L2 penalty). +

                                      • learning_rate (float or callable) – The learning rate.

                                        • +

                                      • momentum (float, optional) – The momentum strength. Default: 0.95

                                        • +

                                      • weight_decay (float, optional) – The weight decay (L2 penalty). Default: 0.01

                                        • -

                                      • nesterov (bool, optional) – Enables Nesterov momentum. Recommended for +

                                      • nesterov (bool, optional) – Enables Nesterov momentum. Recommended for better performance. Default: True

                                        • -

                                      • ns_steps (int, optional) – Number of Newton-Schulz iteration steps for +

                                      • ns_steps (int, optional) – Number of Newton-Schulz iteration steps for orthogonalization. Default: 5

                                      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 b86b1a4a7..41ff92de9 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.29.2 documentation + mlx.optimizers.Optimizer.apply_gradients — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                      mlx.optimizers.Optimizer.apply_gradients#

                                      -Optimizer.apply_gradients(gradients: dict, parameters: dict)#
                                      +Optimizer.apply_gradients(gradients: dict, parameters: dict)#

                                      Apply the gradients to the parameters and return the updated parameters.

                                      Can be used to update a model via model.update(opt.apply_gradients(grads, model)) which is precisely @@ -940,8 +940,8 @@ how

                                      Parameters:
                                        -

                                      • gradients (dict) – A Python tree of gradients.

                                        • -

                                      • parameters (dict) – A Python tree of parameters. It can be a +

                                      • gradients (dict) – A Python tree of gradients.

                                        • +

                                      • parameters (dict) – A Python tree of parameters. It can be a superset of the gradients. In that case the returned python tree will be of the same structure as the gradients.

                                      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 0dac1ef17..ad62bfe67 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.29.2 documentation + mlx.optimizers.Optimizer.init — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,7 +932,7 @@ document.write(`

                                      mlx.optimizers.Optimizer.init#

                                      -Optimizer.init(parameters: dict)#
                                      +Optimizer.init(parameters: dict)#

                                      Initialize the optimizer’s state

                                      This function can be used to initialize optimizers which have state (like momentum in SGD). Using this method is optional as the @@ -942,7 +942,7 @@ to have access to the Optimizer.update().

                                      Parameters:
                                      -

                                      model (dict) – A Python tree of parameters.

                                      +

                                      model (dict) – A Python tree of parameters.

                                      Example

                                      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 28688b690..28a0c8fee 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.29.2 documentation + mlx.optimizers.Optimizer.state — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + 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 cd3289be5..bd82b7604 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.29.2 documentation + mlx.optimizers.Optimizer.update — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                                      mlx.optimizers.Optimizer.update#

                                      -Optimizer.update(model: Module, gradients: dict)#
                                      +Optimizer.update(model: Module, gradients: dict)#

                                      Apply the gradients to the parameters of the model and update the model with the new parameters.

                                      Parameters:

                                      • model (Module) – An mlx module to be updated.

                                        • -

                                      • gradients (dict) – A Python tree of gradients, most likely computed +

                                      • gradients (dict) – A Python tree of gradients, most likely computed via mlx.nn.value_and_grad().

                                      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 d4d104b52..c476028b3 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.29.2 documentation + mlx.optimizers.RMSprop — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,7 +933,7 @@ document.write(`

                                      mlx.optimizers.RMSprop#

                                      -class RMSprop(learning_rate: float | Callable[[array], array], alpha: float = 0.99, eps: float = 1e-08)#
                                      +class RMSprop(learning_rate: float | Callable[[array], array], alpha: float = 0.99, eps: float = 1e-08)#

                                      The RMSprop optimizer [1].

                                      [1]: Tieleman, T. and Hinton, G. 2012. Lecture 6.5-rmsprop, coursera: Neural networks for machine learning

                                      @@ -942,10 +942,10 @@ w_{t+1} &= w_t - \lambda \frac{g_t}{\sqrt{v_{t+1}} + \epsilon}\end{split}\]<
                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • -

                                      • alpha (float, optional) – The smoothing constant \(\alpha\). +

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • +

                                      • alpha (float, optional) – The smoothing constant \(\alpha\). Default: 0.99

                                        • -

                                      • eps (float, optional) – The term \(\epsilon\) added to the denominator +

                                      • eps (float, optional) – The term \(\epsilon\) added to the denominator to improve numerical stability. Default: 1e-8

                                      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 2f7247d52..00045aaf8 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.29.2 documentation + mlx.optimizers.SGD — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                                      @@ -933,7 +933,7 @@ document.write(`

                                      mlx.optimizers.SGD#

                                      -class SGD(learning_rate: float | Callable[[array], array], momentum: float = 0.0, weight_decay: float = 0.0, dampening: float = 0.0, nesterov: bool = False)#
                                      +class SGD(learning_rate: float | Callable[[array], array], momentum: float = 0.0, weight_decay: float = 0.0, dampening: float = 0.0, nesterov: bool = False)#

                                      The stochastic gradient descent optimizer.

                                      Updates a parameter \(w\) with a gradient \(g\) as follows

                                      @@ -942,11 +942,11 @@ w_{t+1} &= w_t - \lambda v_{t+1}\end{split}\]
                                      Parameters:
                                        -

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • -

                                      • momentum (float, optional) – The momentum strength \(\mu\). Default: 0

                                        • -

                                      • weight_decay (float, optional) – The weight decay (L2 penalty). Default: 0

                                        • -

                                      • dampening (float, optional) – Dampening for momentum \(\tau\). Default: 0

                                        • -

                                      • nesterov (bool, optional) – Enables Nesterov momentum. Default: False

                                        • +

                                      • learning_rate (float or callable) – The learning rate \(\lambda\).

                                        • +

                                      • momentum (float, optional) – The momentum strength \(\mu\). Default: 0

                                        • +

                                      • weight_decay (float, optional) – The weight decay (L2 penalty). Default: 0

                                        • +

                                      • dampening (float, optional) – Dampening for momentum \(\tau\). Default: 0

                                        • +

                                      • nesterov (bool, optional) – Enables Nesterov momentum. Default: False

                                      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 e4d80724a..de1e2cdbd 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.29.2 documentation + mlx.optimizers.cosine_decay — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,15 +932,15 @@ document.write(`

                                      mlx.optimizers.cosine_decay#

                                      -cosine_decay(init: float, decay_steps: int, end: float = 0.0) → Callable#
                                      +cosine_decay(init: float, decay_steps: int, end: float = 0.0) → Callable#

                                      Make a cosine decay scheduler.

                                      Parameters:
                                        -

                                      • init (float) – Initial value.

                                        • -

                                      • decay_steps (int) – Number of steps to decay over. The decayed +

                                      • init (float) – Initial value.

                                        • +

                                      • decay_steps (int) – Number of steps to decay over. The decayed value is constant for steps beyond decay_steps.

                                        • -

                                      • end (float, optional) – Final value to decay to. Default: 0.

                                        • +

                                      • end (float, optional) – Final value to decay to. Default: 0.

                                      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 4c86ab065..86c7c3b95 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.29.2 documentation + mlx.optimizers.exponential_decay — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,13 +932,13 @@ document.write(`

                                      mlx.optimizers.exponential_decay#

                                      -exponential_decay(init: float, decay_rate: float) → Callable#
                                      +exponential_decay(init: float, decay_rate: float) → Callable#

                                      Make an exponential decay scheduler.

                                      Parameters:
                                        -

                                      • init (float) – Initial value.

                                        • -

                                      • decay_rate (float) – Multiplicative factor to decay by.

                                        • +

                                      • init (float) – Initial value.

                                        • +

                                      • decay_rate (float) – Multiplicative factor to decay by.

                                      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 f1dc7ecc5..6ad78099c 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.29.2 documentation + mlx.optimizers.join_schedules — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -933,15 +933,15 @@ document.write(`

                                      mlx.optimizers.join_schedules#

                                      -join_schedules(schedules: List[Callable], boundaries: List[int]) → Callable#
                                      +join_schedules(schedules: List[Callable], boundaries: List[int]) → Callable#

                                      Join multiple schedules to create a new schedule.

                                      Parameters:
                                        -

                                      • schedules (list(Callable)) – A list of schedules. Schedule \(i+1\) +

                                      • schedules (list(Callable)) – A list of schedules. Schedule \(i+1\) receives a step count indicating the number of steps since the \(i\)-th boundary.

                                        • -

                                      • boundaries (list(int)) – A list of integers of length len(schedules) - 1 +

                                      • boundaries (list(int)) – A list of integers of length len(schedules) - 1 that indicates when to transition between schedules.

                                      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 65cb716d3..bf2b3fcbf 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.29.2 documentation + mlx.optimizers.linear_schedule — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                                      mlx.optimizers.linear_schedule#

                                      -linear_schedule(init: float, end: float, steps: int) → Callable#
                                      +linear_schedule(init: float, end: float, steps: int) → Callable#

                                      Make a linear scheduler.

                                      Parameters:
                                        -

                                      • init (float) – Initial value.

                                        • -

                                      • end (float) – Final value.

                                        • -

                                      • steps (int) – Number of steps to apply the schedule over. The value is +

                                      • init (float) – Initial value.

                                        • +

                                      • end (float) – Final value.

                                        • +

                                      • steps (int) – Number of steps to apply the schedule over. The value is end for any steps beyond steps.

                                      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 487c69f4b..d3754dccb 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.29.2 documentation + mlx.optimizers.step_decay — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -932,14 +932,14 @@ document.write(`

                                      mlx.optimizers.step_decay#

                                      -step_decay(init: float, decay_rate: float, step_size: int) → Callable#
                                      +step_decay(init: float, decay_rate: float, step_size: int) → Callable#

                                      Make a step decay scheduler.

                                      Parameters:
                                        -

                                      • init (float) – Initial value.

                                        • -

                                      • decay_rate (float) – Multiplicative factor to decay by.

                                        • -

                                      • step_size (int) – Decay every step_size steps.

                                        • +

                                      • init (float) – Initial value.

                                        • +

                                      • decay_rate (float) – Multiplicative factor to decay by.

                                        • +

                                      • step_size (int) – Decay every step_size steps.

                                      diff --git a/docs/build/html/python/optimizers/common_optimizers.html b/docs/build/html/python/optimizers/common_optimizers.html index c869c4ff1..756e18a13 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.29.2 documentation + Common Optimizers — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/optimizers/optimizer.html b/docs/build/html/python/optimizers/optimizer.html index ee6a62624..dd7473e08 100644 --- a/docs/build/html/python/optimizers/optimizer.html +++ b/docs/build/html/python/optimizers/optimizer.html @@ -8,7 +8,7 @@ - Optimizer — MLX 0.29.2 documentation + Optimizer — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/optimizers/schedulers.html b/docs/build/html/python/optimizers/schedulers.html index 609c70bdd..2db501201 100644 --- a/docs/build/html/python/optimizers/schedulers.html +++ b/docs/build/html/python/optimizers/schedulers.html @@ -8,7 +8,7 @@ - Schedulers — MLX 0.29.2 documentation + Schedulers — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/random.html b/docs/build/html/python/random.html index f5993c0cb..778f673b2 100644 --- a/docs/build/html/python/random.html +++ b/docs/build/html/python/random.html @@ -8,7 +8,7 @@ - Random — MLX 0.29.2 documentation + Random — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/transforms.html b/docs/build/html/python/transforms.html index 59499a0d6..692243d72 100644 --- a/docs/build/html/python/transforms.html +++ b/docs/build/html/python/transforms.html @@ -8,7 +8,7 @@ - Transforms — MLX 0.29.2 documentation + Transforms — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/python/tree_utils.html b/docs/build/html/python/tree_utils.html index 74de991d6..ac4ac6e3e 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.29.2 documentation + Tree Utils — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/search.html b/docs/build/html/search.html index 76f686e03..7840544f9 100644 --- a/docs/build/html/search.html +++ b/docs/build/html/search.html @@ -6,7 +6,7 @@ - Search - MLX 0.29.2 documentation + Search - MLX 0.29.3 documentation @@ -35,7 +35,7 @@ - + @@ -141,8 +141,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/searchindex.js b/docs/build/html/searchindex.js index 18a6bb041..3584598c1 100644 --- a/docs/build/html/searchindex.js +++ b/docs/build/html/searchindex.js @@ -1 +1 @@ -Search.setIndex({"alltitles": {"A Simple Example": [[527, "a-simple-example"]], "Array": [[345, null]], "Attention layer": [[6, "attention-layer"]], "Automatic Differentiation": [[520, "automatic-differentiation"]], "Automatic Vectorization": [[520, "automatic-vectorization"]], "Basics": [[525, "basics"]], "Basics of Compile": [[517, "basics-of-compile"]], "Basics of Exporting": [[519, "basics-of-exporting"]], "Binary Size Minimization": [[9, "binary-size-minimization"]], "Binding to Python": [[2, "binding-to-python"]], "Build Options": [[9, "id5"]], "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]], "CPU-only (Linux)": [[9, "cpu-only-linux"]], "CUDA": [[9, "cuda"], [9, "id3"], [346, null]], "Common Optimizers": [[511, null]], "Compilation": [[517, null]], "Compiling Training Graphs": [[517, "compiling-training-graphs"]], "Complex Example": [[1, "complex-example"]], "Conversion to NumPy and Other Frameworks": [[524, null]], "Converting the weights": [[6, "converting-the-weights"]], "Custom Extensions in MLX": [[2, null]], "Custom Metal Kernels": [[1, null]], "Data Types": [[347, null]], "Debugging": [[517, "debugging"]], "Defining a Ring": [[518, "defining-a-ring"]], "Devices and Streams": [[348, null]], "Differences from NumPy": [[521, "differences-from-numpy"]], "Distributed Communication": [[349, null], [518, null]], "Download the code": [[2, null], [6, null]], "Encoder layer": [[6, "encoder-layer"]], "Example Speedup": [[517, "example-speedup"]], "Examples": [[8, null]], "Export Functions": [[350, null]], "Exporting Functions": [[519, null]], "Exporting Modules": [[519, "exporting-modules"]], "Exporting Multiple Traces": [[519, "exporting-multiple-traces"]], "FFT": [[352, null]], "Fast": [[351, null]], "Full model": [[6, "full-model"]], "Function Transforms": [[520, null]], "Function and Graph Transformations": [[525, "function-and-graph-transformations"]], "Functions": [[484, null]], "Further Reading": [[8, null]], "Generation": [[6, "generation"]], "Getting Started": [[518, "getting-started"]], "Getting Started with MPI": [[518, "getting-started-with-mpi"]], "Getting Started with Ring": [[518, "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++": [[519, "importing-functions-in-c"]], "In Place Updates": [[521, "in-place-updates"]], "Indexing Arrays": [[521, null]], "Initializers": [[485, null]], "Inspecting Modules": [[356, "inspecting-modules"]], "Install": [[8, null]], "Installing MPI": [[518, "installing-mpi"]], "Introducing the Example": [[2, "introducing-the-example"]], "JAX": [[524, "jax"]], "LLM inference": [[6, null]], "Launching Distributed Programs": [[522, null]], "Layers": [[486, null]], "Lazy Evaluation": [[523, null]], "Linear Algebra": [[353, null]], "Linear Regression": [[5, null]], "Linux": [[9, "linux"]], "Loss Functions": [[487, null]], "MLX": [[8, null]], "MPI Specifics": [[522, "mpi-specifics"]], "Memory Management": [[354, null]], "Metal": [[355, null]], "Metal Debugger": [[3, null]], "Metal not found": [[9, "metal-not-found"]], "Module": [[488, null]], "More Examples": [[519, "more-examples"]], "Multi-Layer Perceptron": [[7, null]], "Neural Networks": [[356, null]], "Only Compute What You Use": [[523, "only-compute-what-you-use"]], "Operations": [[0, null], [2, "operations"], [489, null]], "Operations and Primitives": [[2, "operations-and-primitives"]], "Optimizer": [[512, null]], "Optimizers": [[490, null]], "Package Variables": [[4, "id1"]], "Parameters": [[356, "parameters"]], "Primitive Transforms": [[2, "primitive-transforms"]], "Primitives": [[2, "primitives"]], "Providing Hosts": [[522, "providing-hosts"]], "Pure Functions": [[517, "pure-functions"]], "Putting it all together": [[6, "putting-it-all-together"]], "PyTorch": [[524, "pytorch"]], "Python API": [[9, "python-api"]], "Python API Reference": [[8, null]], "Python Installation": [[9, "python-installation"]], "Quick Start Guide": [[525, null]], "Quick Start with Neural Networks": [[356, "quick-start-with-neural-networks"]], "Random": [[514, null]], "Results": [[2, "results"]], "Ring Specifics": [[522, "ring-specifics"]], "Running Distributed Programs": [[518, "running-distributed-programs"]], "Saving and Loading": [[490, "saving-and-loading"]], "Saving and Loading Arrays": [[526, null]], "Schedulers": [[513, null]], "Scripts": [[2, "scripts"], [6, "scripts"]], "Selecting Backend": [[518, "selecting-backend"]], "Serialization Formats": [[526, "id1"]], "Setting up Remote Hosts": [[518, "setting-up-remote-hosts"], [522, "setting-up-remote-hosts"]], "Shapeless Compilation": [[517, "shapeless-compilation"]], "Shapeless Exports": [[519, "shapeless-exports"]], "Simple Example": [[1, "simple-example"]], "Specifying the Stream": [[528, "specifying-the-stream"]], "Supported Data Types": [[347, "id2"]], "TensorFlow": [[524, "tensorflow"]], "The Module Class": [[356, "the-module-class"]], "Thunderbolt Ring": [[518, "thunderbolt-ring"]], "Training Example": [[518, "training-example"]], "Transformations with Compile": [[517, "transformations-with-compile"]], "Transformations with Imported Functions": [[519, "transformations-with-imported-functions"]], "Transforming Compute Graphs": [[523, "transforming-compute-graphs"]], "Transforms": [[515, null]], "Tree Utils": [[516, null]], "Troubleshooting": [[9, "troubleshooting"], [9, "id4"]], "Tuning MPI All Reduce": [[518, "tuning-mpi-all-reduce"]], "Unified Memory": [[527, null]], "Updating the Parameters": [[356, "updating-the-parameters"]], "Usage": [[2, "usage"], [8, null], [522, "usage"]], "Using MLX in C++": [[4, null]], "Using Shape/Strides": [[1, "using-shape-strides"]], "Using Streams": [[528, null]], "Using the Primitive": [[2, "using-the-primitive"]], "Utilizing nn.average_gradients": [[518, "utilizing-nn-average-gradients"]], "Value and Grad": [[356, "value-and-grad"]], "Weight loading and benchmarking": [[6, "weight-loading-and-benchmarking"]], "When to Evaluate": [[523, "when-to-evaluate"]], "Why Lazy Evaluation": [[523, "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": [[344, 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.imag": [[51, null]], "mlx.core.array.item": [[52, null]], "mlx.core.array.itemsize": [[53, null]], "mlx.core.array.log": [[54, null]], "mlx.core.array.log10": [[55, null]], "mlx.core.array.log1p": [[56, null]], "mlx.core.array.log2": [[57, null]], "mlx.core.array.logcumsumexp": [[58, null]], "mlx.core.array.logsumexp": [[59, null]], "mlx.core.array.max": [[60, null]], "mlx.core.array.mean": [[61, null]], "mlx.core.array.min": [[62, null]], "mlx.core.array.moveaxis": [[63, null]], "mlx.core.array.nbytes": [[64, null]], "mlx.core.array.ndim": [[65, null]], "mlx.core.array.prod": [[66, null]], "mlx.core.array.real": [[67, null]], "mlx.core.array.reciprocal": [[68, null]], "mlx.core.array.reshape": [[69, null]], "mlx.core.array.round": [[70, null]], "mlx.core.array.rsqrt": [[71, null]], "mlx.core.array.shape": [[72, null]], "mlx.core.array.sin": [[73, null]], "mlx.core.array.size": [[74, null]], "mlx.core.array.split": [[75, null]], "mlx.core.array.sqrt": [[76, null]], "mlx.core.array.square": [[77, null]], "mlx.core.array.squeeze": [[78, null]], "mlx.core.array.std": [[79, null]], "mlx.core.array.sum": [[80, null]], "mlx.core.array.swapaxes": [[81, null]], "mlx.core.array.tolist": [[82, null]], "mlx.core.array.transpose": [[83, null]], "mlx.core.array.var": [[84, null]], "mlx.core.array.view": [[85, null]], "mlx.core.array_equal": [[86, null]], "mlx.core.as_strided": [[87, null]], "mlx.core.async_eval": [[88, null]], "mlx.core.atleast_1d": [[89, null]], "mlx.core.atleast_2d": [[90, null]], "mlx.core.atleast_3d": [[91, null]], "mlx.core.bitwise_and": [[92, null]], "mlx.core.bitwise_invert": [[93, null]], "mlx.core.bitwise_or": [[94, null]], "mlx.core.bitwise_xor": [[95, null]], "mlx.core.block_masked_mm": [[96, null]], "mlx.core.broadcast_arrays": [[97, null]], "mlx.core.broadcast_to": [[98, null]], "mlx.core.ceil": [[99, null]], "mlx.core.clear_cache": [[100, null]], "mlx.core.clip": [[101, null]], "mlx.core.compile": [[102, null]], "mlx.core.concatenate": [[103, null]], "mlx.core.conj": [[104, null]], "mlx.core.conjugate": [[105, null]], "mlx.core.contiguous": [[106, null]], "mlx.core.conv1d": [[107, null]], "mlx.core.conv2d": [[108, null]], "mlx.core.conv3d": [[109, null]], "mlx.core.conv_general": [[110, null]], "mlx.core.conv_transpose1d": [[111, null]], "mlx.core.conv_transpose2d": [[112, null]], "mlx.core.conv_transpose3d": [[113, null]], "mlx.core.convolve": [[114, null]], "mlx.core.cos": [[115, null]], "mlx.core.cosh": [[116, null]], "mlx.core.cuda.is_available": [[117, null]], "mlx.core.cummax": [[118, null]], "mlx.core.cummin": [[119, null]], "mlx.core.cumprod": [[120, null]], "mlx.core.cumsum": [[121, null]], "mlx.core.custom_function": [[122, null]], "mlx.core.default_device": [[123, null]], "mlx.core.default_stream": [[124, null]], "mlx.core.degrees": [[125, null]], "mlx.core.dequantize": [[126, null]], "mlx.core.diag": [[127, null]], "mlx.core.diagonal": [[128, null]], "mlx.core.disable_compile": [[129, null]], "mlx.core.distributed.Group": [[130, null]], "mlx.core.distributed.all_gather": [[131, null]], "mlx.core.distributed.all_sum": [[132, null]], "mlx.core.distributed.init": [[133, null]], "mlx.core.distributed.is_available": [[134, null]], "mlx.core.distributed.recv": [[135, null]], "mlx.core.distributed.recv_like": [[136, null]], "mlx.core.distributed.send": [[137, null]], "mlx.core.divide": [[138, null]], "mlx.core.divmod": [[139, null]], "mlx.core.einsum": [[140, null]], "mlx.core.einsum_path": [[141, null]], "mlx.core.enable_compile": [[142, null]], "mlx.core.equal": [[143, null]], "mlx.core.erf": [[144, null]], "mlx.core.erfinv": [[145, null]], "mlx.core.eval": [[146, null]], "mlx.core.exp": [[147, null]], "mlx.core.expand_dims": [[148, null]], "mlx.core.expm1": [[149, null]], "mlx.core.export_function": [[150, null]], "mlx.core.export_to_dot": [[151, null]], "mlx.core.exporter": [[152, null]], "mlx.core.eye": [[153, null]], "mlx.core.fast.cuda_kernel": [[154, null]], "mlx.core.fast.layer_norm": [[155, null]], "mlx.core.fast.metal_kernel": [[156, null]], "mlx.core.fast.rms_norm": [[157, null]], "mlx.core.fast.rope": [[158, null]], "mlx.core.fast.scaled_dot_product_attention": [[159, null]], "mlx.core.fft.fft": [[160, null]], "mlx.core.fft.fft2": [[161, null]], "mlx.core.fft.fftn": [[162, null]], "mlx.core.fft.fftshift": [[163, null]], "mlx.core.fft.ifft": [[164, null]], "mlx.core.fft.ifft2": [[165, null]], "mlx.core.fft.ifftn": [[166, null]], "mlx.core.fft.ifftshift": [[167, null]], "mlx.core.fft.irfft": [[168, null]], "mlx.core.fft.irfft2": [[169, null]], "mlx.core.fft.irfftn": [[170, null]], "mlx.core.fft.rfft": [[171, null]], "mlx.core.fft.rfft2": [[172, null]], "mlx.core.fft.rfftn": [[173, null]], "mlx.core.finfo": [[174, null]], "mlx.core.flatten": [[175, null]], "mlx.core.floor": [[176, null]], "mlx.core.floor_divide": [[177, null]], "mlx.core.full": [[178, null]], "mlx.core.gather_mm": [[179, null]], "mlx.core.gather_qmm": [[180, null]], "mlx.core.get_active_memory": [[181, null]], "mlx.core.get_cache_memory": [[182, null]], "mlx.core.get_peak_memory": [[183, null]], "mlx.core.grad": [[184, null]], "mlx.core.greater": [[185, null]], "mlx.core.greater_equal": [[186, null]], "mlx.core.hadamard_transform": [[187, null]], "mlx.core.identity": [[188, null]], "mlx.core.imag": [[189, null]], "mlx.core.import_function": [[190, null]], "mlx.core.inner": [[191, null]], "mlx.core.isclose": [[192, null]], "mlx.core.isfinite": [[193, null]], "mlx.core.isinf": [[194, null]], "mlx.core.isnan": [[195, null]], "mlx.core.isneginf": [[196, null]], "mlx.core.isposinf": [[197, null]], "mlx.core.issubdtype": [[198, null]], "mlx.core.jvp": [[199, null]], "mlx.core.kron": [[200, null]], "mlx.core.left_shift": [[201, null]], "mlx.core.less": [[202, null]], "mlx.core.less_equal": [[203, null]], "mlx.core.linalg.cholesky": [[204, null]], "mlx.core.linalg.cholesky_inv": [[205, null]], "mlx.core.linalg.cross": [[206, null]], "mlx.core.linalg.eig": [[207, null]], "mlx.core.linalg.eigh": [[208, null]], "mlx.core.linalg.eigvals": [[209, null]], "mlx.core.linalg.eigvalsh": [[210, null]], "mlx.core.linalg.inv": [[211, null]], "mlx.core.linalg.lu": [[212, null]], "mlx.core.linalg.lu_factor": [[213, null]], "mlx.core.linalg.norm": [[214, null]], "mlx.core.linalg.pinv": [[215, null]], "mlx.core.linalg.qr": [[216, null]], "mlx.core.linalg.solve": [[217, null]], "mlx.core.linalg.solve_triangular": [[218, null]], "mlx.core.linalg.svd": [[219, null]], "mlx.core.linalg.tri_inv": [[220, null]], "mlx.core.linspace": [[221, null]], "mlx.core.load": [[222, null]], "mlx.core.log": [[223, null]], "mlx.core.log10": [[224, null]], "mlx.core.log1p": [[225, null]], "mlx.core.log2": [[226, null]], "mlx.core.logaddexp": [[227, null]], "mlx.core.logcumsumexp": [[228, null]], "mlx.core.logical_and": [[229, null]], "mlx.core.logical_not": [[230, null]], "mlx.core.logical_or": [[231, null]], "mlx.core.logsumexp": [[232, null]], "mlx.core.matmul": [[233, null]], "mlx.core.max": [[234, null]], "mlx.core.maximum": [[235, null]], "mlx.core.mean": [[236, null]], "mlx.core.meshgrid": [[237, null]], "mlx.core.metal.device_info": [[238, null]], "mlx.core.metal.is_available": [[239, null]], "mlx.core.metal.start_capture": [[240, null]], "mlx.core.metal.stop_capture": [[241, null]], "mlx.core.min": [[242, null]], "mlx.core.minimum": [[243, null]], "mlx.core.moveaxis": [[244, null]], "mlx.core.multiply": [[245, null]], "mlx.core.nan_to_num": [[246, null]], "mlx.core.negative": [[247, null]], "mlx.core.new_stream": [[248, null]], "mlx.core.not_equal": [[249, null]], "mlx.core.ones": [[250, null]], "mlx.core.ones_like": [[251, null]], "mlx.core.outer": [[252, null]], "mlx.core.pad": [[253, null]], "mlx.core.partition": [[254, null]], "mlx.core.power": [[255, null]], "mlx.core.prod": [[256, null]], "mlx.core.put_along_axis": [[257, null]], "mlx.core.quantize": [[258, null]], "mlx.core.quantized_matmul": [[259, null]], "mlx.core.radians": [[260, null]], "mlx.core.random.bernoulli": [[261, null]], "mlx.core.random.categorical": [[262, null]], "mlx.core.random.gumbel": [[263, null]], "mlx.core.random.key": [[264, null]], "mlx.core.random.laplace": [[265, null]], "mlx.core.random.multivariate_normal": [[266, null]], "mlx.core.random.normal": [[267, null]], "mlx.core.random.permutation": [[268, null]], "mlx.core.random.randint": [[269, null]], "mlx.core.random.seed": [[270, null]], "mlx.core.random.split": [[271, null]], "mlx.core.random.truncated_normal": [[272, null]], "mlx.core.random.uniform": [[273, null]], "mlx.core.real": [[274, null]], "mlx.core.reciprocal": [[275, null]], "mlx.core.remainder": [[276, null]], "mlx.core.repeat": [[277, null]], "mlx.core.reset_peak_memory": [[278, null]], "mlx.core.reshape": [[279, null]], "mlx.core.right_shift": [[280, null]], "mlx.core.roll": [[281, null]], "mlx.core.round": [[282, null]], "mlx.core.rsqrt": [[283, null]], "mlx.core.save": [[284, null]], "mlx.core.save_gguf": [[285, null]], "mlx.core.save_safetensors": [[286, null]], "mlx.core.savez": [[287, null]], "mlx.core.savez_compressed": [[288, null]], "mlx.core.set_cache_limit": [[289, null]], "mlx.core.set_default_device": [[290, null]], "mlx.core.set_default_stream": [[291, null]], "mlx.core.set_memory_limit": [[292, null]], "mlx.core.set_wired_limit": [[293, null]], "mlx.core.sigmoid": [[294, null]], "mlx.core.sign": [[295, null]], "mlx.core.sin": [[296, null]], "mlx.core.sinh": [[297, null]], "mlx.core.slice": [[298, null]], "mlx.core.slice_update": [[299, null]], "mlx.core.softmax": [[300, null]], "mlx.core.sort": [[301, null]], "mlx.core.split": [[302, null]], "mlx.core.sqrt": [[303, null]], "mlx.core.square": [[304, null]], "mlx.core.squeeze": [[305, null]], "mlx.core.stack": [[306, null]], "mlx.core.std": [[307, null]], "mlx.core.stop_gradient": [[308, null]], "mlx.core.stream": [[309, null]], "mlx.core.subtract": [[310, null]], "mlx.core.sum": [[311, null]], "mlx.core.swapaxes": [[312, null]], "mlx.core.synchronize": [[313, null]], "mlx.core.take": [[314, null]], "mlx.core.take_along_axis": [[315, null]], "mlx.core.tan": [[316, null]], "mlx.core.tanh": [[317, null]], "mlx.core.tensordot": [[318, null]], "mlx.core.tile": [[319, null]], "mlx.core.topk": [[320, null]], "mlx.core.trace": [[321, null]], "mlx.core.transpose": [[322, null]], "mlx.core.tri": [[323, null]], "mlx.core.tril": [[324, null]], "mlx.core.triu": [[325, null]], "mlx.core.unflatten": [[326, null]], "mlx.core.value_and_grad": [[327, null]], "mlx.core.var": [[328, null]], "mlx.core.view": [[329, null]], "mlx.core.vjp": [[330, null]], "mlx.core.vmap": [[331, null]], "mlx.core.where": [[332, null]], "mlx.core.zeros": [[333, null]], "mlx.core.zeros_like": [[334, null]], "mlx.nn.ALiBi": [[357, null]], "mlx.nn.AvgPool1d": [[358, null]], "mlx.nn.AvgPool2d": [[359, null]], "mlx.nn.AvgPool3d": [[360, null]], "mlx.nn.BatchNorm": [[361, null]], "mlx.nn.CELU": [[362, null]], "mlx.nn.Conv1d": [[363, null]], "mlx.nn.Conv2d": [[364, null]], "mlx.nn.Conv3d": [[365, null]], "mlx.nn.ConvTranspose1d": [[366, null]], "mlx.nn.ConvTranspose2d": [[367, null]], "mlx.nn.ConvTranspose3d": [[368, null]], "mlx.nn.Dropout": [[369, null]], "mlx.nn.Dropout2d": [[370, null]], "mlx.nn.Dropout3d": [[371, null]], "mlx.nn.ELU": [[372, null]], "mlx.nn.Embedding": [[373, null]], "mlx.nn.GELU": [[374, null]], "mlx.nn.GLU": [[375, null]], "mlx.nn.GRU": [[376, null]], "mlx.nn.GroupNorm": [[377, null]], "mlx.nn.HardShrink": [[378, null]], "mlx.nn.HardTanh": [[379, null]], "mlx.nn.Hardswish": [[380, null]], "mlx.nn.InstanceNorm": [[381, null]], "mlx.nn.LSTM": [[382, null]], "mlx.nn.LayerNorm": [[383, null]], "mlx.nn.LeakyReLU": [[384, null]], "mlx.nn.Linear": [[385, null]], "mlx.nn.LogSigmoid": [[386, null]], "mlx.nn.LogSoftmax": [[387, null]], "mlx.nn.MaxPool1d": [[388, null]], "mlx.nn.MaxPool2d": [[389, null]], "mlx.nn.MaxPool3d": [[390, null]], "mlx.nn.Mish": [[391, null]], "mlx.nn.Module.apply": [[392, null]], "mlx.nn.Module.apply_to_modules": [[393, null]], "mlx.nn.Module.children": [[394, null]], "mlx.nn.Module.eval": [[395, null]], "mlx.nn.Module.filter_and_map": [[396, null]], "mlx.nn.Module.freeze": [[397, null]], "mlx.nn.Module.leaf_modules": [[398, null]], "mlx.nn.Module.load_weights": [[399, null]], "mlx.nn.Module.modules": [[400, null]], "mlx.nn.Module.named_modules": [[401, null]], "mlx.nn.Module.parameters": [[402, null]], "mlx.nn.Module.save_weights": [[403, null]], "mlx.nn.Module.set_dtype": [[404, null]], "mlx.nn.Module.state": [[405, null]], "mlx.nn.Module.train": [[406, null]], "mlx.nn.Module.trainable_parameters": [[407, null]], "mlx.nn.Module.training": [[408, null]], "mlx.nn.Module.unfreeze": [[409, null]], "mlx.nn.Module.update": [[410, null]], "mlx.nn.Module.update_modules": [[411, null]], "mlx.nn.MultiHeadAttention": [[412, null]], "mlx.nn.PReLU": [[413, null]], "mlx.nn.QuantizedEmbedding": [[414, null]], "mlx.nn.QuantizedLinear": [[415, null]], "mlx.nn.RMSNorm": [[416, null]], "mlx.nn.RNN": [[417, null]], "mlx.nn.ReLU": [[418, null]], "mlx.nn.ReLU2": [[419, null]], "mlx.nn.ReLU6": [[420, null]], "mlx.nn.RoPE": [[421, null]], "mlx.nn.SELU": [[422, null]], "mlx.nn.Sequential": [[423, null]], "mlx.nn.SiLU": [[424, null]], "mlx.nn.Sigmoid": [[425, null]], "mlx.nn.SinusoidalPositionalEncoding": [[426, null]], "mlx.nn.Softmax": [[427, null]], "mlx.nn.Softmin": [[428, null]], "mlx.nn.Softplus": [[429, null]], "mlx.nn.Softshrink": [[430, null]], "mlx.nn.Softsign": [[431, null]], "mlx.nn.Step": [[432, null]], "mlx.nn.Tanh": [[433, null]], "mlx.nn.Transformer": [[434, null]], "mlx.nn.Upsample": [[435, null]], "mlx.nn.average_gradients": [[335, null]], "mlx.nn.celu": [[444, null]], "mlx.nn.elu": [[445, null]], "mlx.nn.gelu": [[446, null]], "mlx.nn.gelu_approx": [[447, null]], "mlx.nn.gelu_fast_approx": [[448, null]], "mlx.nn.glu": [[449, null]], "mlx.nn.hard_shrink": [[450, null]], "mlx.nn.hard_tanh": [[451, null]], "mlx.nn.hardswish": [[452, null]], "mlx.nn.init.constant": [[436, null]], "mlx.nn.init.glorot_normal": [[437, null]], "mlx.nn.init.glorot_uniform": [[438, null]], "mlx.nn.init.he_normal": [[439, null]], "mlx.nn.init.he_uniform": [[440, null]], "mlx.nn.init.identity": [[441, null]], "mlx.nn.init.normal": [[442, null]], "mlx.nn.init.uniform": [[443, null]], "mlx.nn.leaky_relu": [[453, null]], "mlx.nn.log_sigmoid": [[454, null]], "mlx.nn.log_softmax": [[455, null]], "mlx.nn.losses.binary_cross_entropy": [[456, null]], "mlx.nn.losses.cosine_similarity_loss": [[457, null]], "mlx.nn.losses.cross_entropy": [[458, null]], "mlx.nn.losses.gaussian_nll_loss": [[459, null]], "mlx.nn.losses.hinge_loss": [[460, null]], "mlx.nn.losses.huber_loss": [[461, null]], "mlx.nn.losses.kl_div_loss": [[462, null]], "mlx.nn.losses.l1_loss": [[463, null]], "mlx.nn.losses.log_cosh_loss": [[464, null]], "mlx.nn.losses.margin_ranking_loss": [[465, null]], "mlx.nn.losses.mse_loss": [[466, null]], "mlx.nn.losses.nll_loss": [[467, null]], "mlx.nn.losses.smooth_l1_loss": [[468, null]], "mlx.nn.losses.triplet_loss": [[469, null]], "mlx.nn.mish": [[470, null]], "mlx.nn.prelu": [[471, null]], "mlx.nn.quantize": [[336, null]], "mlx.nn.relu": [[472, null]], "mlx.nn.relu2": [[473, null]], "mlx.nn.relu6": [[474, null]], "mlx.nn.selu": [[475, null]], "mlx.nn.sigmoid": [[476, null]], "mlx.nn.silu": [[477, null]], "mlx.nn.softmax": [[478, null]], "mlx.nn.softmin": [[479, null]], "mlx.nn.softplus": [[480, null]], "mlx.nn.softshrink": [[481, null]], "mlx.nn.step": [[482, null]], "mlx.nn.tanh": [[483, null]], "mlx.nn.value_and_grad": [[337, null]], "mlx.optimizers.AdaDelta": [[491, null]], "mlx.optimizers.Adafactor": [[492, null]], "mlx.optimizers.Adagrad": [[493, null]], "mlx.optimizers.Adam": [[494, null]], "mlx.optimizers.AdamW": [[495, null]], "mlx.optimizers.Adamax": [[496, null]], "mlx.optimizers.Lion": [[497, null]], "mlx.optimizers.MultiOptimizer": [[498, null]], "mlx.optimizers.Muon": [[499, null]], "mlx.optimizers.Optimizer.apply_gradients": [[500, null]], "mlx.optimizers.Optimizer.init": [[501, null]], "mlx.optimizers.Optimizer.state": [[502, null]], "mlx.optimizers.Optimizer.update": [[503, null]], "mlx.optimizers.RMSprop": [[504, null]], "mlx.optimizers.SGD": [[505, null]], "mlx.optimizers.clip_grad_norm": [[338, null]], "mlx.optimizers.cosine_decay": [[506, null]], "mlx.optimizers.exponential_decay": [[507, null]], "mlx.optimizers.join_schedules": [[508, null]], "mlx.optimizers.linear_schedule": [[509, null]], "mlx.optimizers.step_decay": [[510, null]], "mlx.utils.tree_flatten": [[339, null]], "mlx.utils.tree_map": [[340, null]], "mlx.utils.tree_map_with_path": [[341, null]], "mlx.utils.tree_reduce": [[342, null]], "mlx.utils.tree_unflatten": [[343, 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.imag", "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.logcumsumexp", "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.real", "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.cuda.is_available", "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.cuda_kernel", "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.fftshift", "python/_autosummary/mlx.core.fft.ifft", "python/_autosummary/mlx.core.fft.ifft2", "python/_autosummary/mlx.core.fft.ifftn", "python/_autosummary/mlx.core.fft.ifftshift", "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.eig", "python/_autosummary/mlx.core.linalg.eigh", "python/_autosummary/mlx.core.linalg.eigvals", "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.logcumsumexp", "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/cuda", "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.ReLU2", "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.relu2", "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.MultiOptimizer", "python/optimizers/_autosummary/mlx.optimizers.Muon", "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.imag.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.logcumsumexp.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.real.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.cuda.is_available.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.cuda_kernel.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.fftshift.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.ifftshift.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.eig.rst", "python/_autosummary/mlx.core.linalg.eigh.rst", "python/_autosummary/mlx.core.linalg.eigvals.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.logcumsumexp.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/cuda.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.ReLU2.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.relu2.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.MultiOptimizer.rst", "python/optimizers/_autosummary/mlx.optimizers.Muon.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)": [[122, "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)": [[174, "mlx.core.finfo.__init__", false]], "__init__() (group method)": [[130, "mlx.core.distributed.Group.__init__", false]], "__init__() (stream method)": [[344, "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)": [[491, "mlx.optimizers.AdaDelta", false]], "adafactor (class in mlx.optimizers)": [[492, "mlx.optimizers.Adafactor", false]], "adagrad (class in mlx.optimizers)": [[493, "mlx.optimizers.Adagrad", false]], "adam (class in mlx.optimizers)": [[494, "mlx.optimizers.Adam", false]], "adamax (class in mlx.optimizers)": [[496, "mlx.optimizers.Adamax", false]], "adamw (class in mlx.optimizers)": [[495, "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)": [[357, "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)": [[131, "mlx.core.distributed.all_gather", false]], "all_sum() (in module mlx.core.distributed)": [[132, "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)": [[392, "mlx.nn.Module.apply", false]], "apply_gradients() (optimizer method)": [[500, "mlx.optimizers.Optimizer.apply_gradients", false]], "apply_to_modules() (module method)": [[393, "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)": [[86, "mlx.core.array_equal", false]], "as_strided (c++ function)": [[0, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", false]], "as_strided() (in module mlx.core)": [[87, "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)": [[88, "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)": [[89, "mlx.core.atleast_1d", false]], "atleast_2d (c++ function)": [[0, "_CPPv410atleast_2dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_2d() (in module mlx.core)": [[90, "mlx.core.atleast_2d", false]], "atleast_3d (c++ function)": [[0, "_CPPv410atleast_3dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_3d() (in module mlx.core)": [[91, "mlx.core.atleast_3d", false]], "average_gradients() (in module mlx.nn)": [[335, "mlx.nn.average_gradients", false]], "avgpool1d (class in mlx.nn)": [[358, "mlx.nn.AvgPool1d", false]], "avgpool2d (class in mlx.nn)": [[359, "mlx.nn.AvgPool2d", false]], "avgpool3d (class in mlx.nn)": [[360, "mlx.nn.AvgPool3d", false]], "batchnorm (class in mlx.nn)": [[361, "mlx.nn.BatchNorm", false]], "bernoulli() (in module mlx.core.random)": [[261, "mlx.core.random.bernoulli", false]], "binary_cross_entropy (class in mlx.nn.losses)": [[456, "mlx.nn.losses.binary_cross_entropy", false]], "bitwise_and (c++ function)": [[0, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", false]], "bitwise_and() (in module mlx.core)": [[92, "mlx.core.bitwise_and", false]], "bitwise_invert (c++ function)": [[0, "_CPPv414bitwise_invertRK5array14StreamOrDevice", false]], "bitwise_invert() (in module mlx.core)": [[93, "mlx.core.bitwise_invert", false]], "bitwise_or (c++ function)": [[0, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", false]], "bitwise_or() (in module mlx.core)": [[94, "mlx.core.bitwise_or", false]], "bitwise_xor (c++ function)": [[0, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", false]], "bitwise_xor() (in module mlx.core)": [[95, "mlx.core.bitwise_xor", false]], "block_masked_mm (c++ function)": [[0, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "block_masked_mm() (in module mlx.core)": [[96, "mlx.core.block_masked_mm", false]], "broadcast_arrays (c++ function)": [[0, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "broadcast_arrays() (in module mlx.core)": [[97, "mlx.core.broadcast_arrays", false]], "broadcast_to (c++ function)": [[0, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", false]], "broadcast_to() (in module mlx.core)": [[98, "mlx.core.broadcast_to", false]], "categorical() (in module mlx.core.random)": [[262, "mlx.core.random.categorical", false]], "ceil (c++ function)": [[0, "_CPPv44ceilRK5array14StreamOrDevice", false]], "ceil() (in module mlx.core)": [[99, "mlx.core.ceil", false]], "celu (class in mlx.nn)": [[362, "mlx.nn.CELU", false], [444, "mlx.nn.celu", false]], "children() (module method)": [[394, "mlx.nn.Module.children", false]], "cholesky() (in module mlx.core.linalg)": [[204, "mlx.core.linalg.cholesky", false]], "cholesky_inv() (in module mlx.core.linalg)": [[205, "mlx.core.linalg.cholesky_inv", false]], "clear_cache() (in module mlx.core)": [[100, "mlx.core.clear_cache", false]], "clip (c++ function)": [[0, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", false]], "clip() (in module mlx.core)": [[101, "mlx.core.clip", false]], "clip_grad_norm() (in module mlx.optimizers)": [[338, "mlx.optimizers.clip_grad_norm", false]], "compile() (in module mlx.core)": [[102, "mlx.core.compile", false]], "concatenate (c++ function)": [[0, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", false]], "concatenate() (in module mlx.core)": [[103, "mlx.core.concatenate", false]], "conj() (array method)": [[40, "mlx.core.array.conj", false]], "conj() (in module mlx.core)": [[104, "mlx.core.conj", false]], "conjugate (c++ function)": [[0, "_CPPv49conjugateRK5array14StreamOrDevice", false]], "conjugate() (in module mlx.core)": [[105, "mlx.core.conjugate", false]], "constant() (in module mlx.nn.init)": [[436, "mlx.nn.init.constant", false]], "contiguous (c++ function)": [[0, "_CPPv410contiguousRK5arrayb14StreamOrDevice", false]], "contiguous() (in module mlx.core)": [[106, "mlx.core.contiguous", false]], "conv1d (c++ function)": [[0, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv1d (class in mlx.nn)": [[363, "mlx.nn.Conv1d", false]], "conv1d() (in module mlx.core)": [[107, "mlx.core.conv1d", false]], "conv2d (c++ function)": [[0, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv2d (class in mlx.nn)": [[364, "mlx.nn.Conv2d", false]], "conv2d() (in module mlx.core)": [[108, "mlx.core.conv2d", false]], "conv3d (c++ function)": [[0, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv3d (class in mlx.nn)": [[365, "mlx.nn.Conv3d", false]], "conv3d() (in module mlx.core)": [[109, "mlx.core.conv3d", false]], "conv_general (c++ function)": [[0, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false], [0, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false]], "conv_general() (in module mlx.core)": [[110, "mlx.core.conv_general", false]], "conv_transpose1d (c++ function)": [[0, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", false]], "conv_transpose1d() (in module mlx.core)": [[111, "mlx.core.conv_transpose1d", false]], "conv_transpose2d (c++ function)": [[0, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv_transpose2d() (in module mlx.core)": [[112, "mlx.core.conv_transpose2d", false]], "conv_transpose3d (c++ function)": [[0, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv_transpose3d() (in module mlx.core)": [[113, "mlx.core.conv_transpose3d", false]], "convolve() (in module mlx.core)": [[114, "mlx.core.convolve", false]], "convtranspose1d (class in mlx.nn)": [[366, "mlx.nn.ConvTranspose1d", false]], "convtranspose2d (class in mlx.nn)": [[367, "mlx.nn.ConvTranspose2d", false]], "convtranspose3d (class in mlx.nn)": [[368, "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)": [[115, "mlx.core.cos", false]], "cosh (c++ function)": [[0, "_CPPv44coshRK5array14StreamOrDevice", false]], "cosh() (in module mlx.core)": [[116, "mlx.core.cosh", false]], "cosine_decay() (in module mlx.optimizers)": [[506, "mlx.optimizers.cosine_decay", false]], "cosine_similarity_loss (class in mlx.nn.losses)": [[457, "mlx.nn.losses.cosine_similarity_loss", false]], "cross() (in module mlx.core.linalg)": [[206, "mlx.core.linalg.cross", false]], "cross_entropy (class in mlx.nn.losses)": [[458, "mlx.nn.losses.cross_entropy", false]], "cuda_kernel() (in module mlx.core.fast)": [[154, "mlx.core.fast.cuda_kernel", false]], "cummax (c++ function)": [[0, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", false]], "cummax() (array method)": [[42, "mlx.core.array.cummax", false]], "cummax() (in module mlx.core)": [[118, "mlx.core.cummax", false]], "cummin (c++ function)": [[0, "_CPPv46cumminRK5arrayibb14StreamOrDevice", false]], "cummin() (array method)": [[43, "mlx.core.array.cummin", false]], "cummin() (in module mlx.core)": [[119, "mlx.core.cummin", false]], "cumprod (c++ function)": [[0, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", false]], "cumprod() (array method)": [[44, "mlx.core.array.cumprod", false]], "cumprod() (in module mlx.core)": [[120, "mlx.core.cumprod", false]], "cumsum (c++ function)": [[0, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", false]], "cumsum() (array method)": [[45, "mlx.core.array.cumsum", false]], "cumsum() (in module mlx.core)": [[121, "mlx.core.cumsum", false]], "custom_function (class in mlx.core)": [[122, "mlx.core.custom_function", false]], "default_device() (in module mlx.core)": [[123, "mlx.core.default_device", false]], "default_stream() (in module mlx.core)": [[124, "mlx.core.default_stream", false]], "degrees (c++ function)": [[0, "_CPPv47degreesRK5array14StreamOrDevice", false]], "degrees() (in module mlx.core)": [[125, "mlx.core.degrees", false]], "depends (c++ function)": [[0, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", false]], "dequantize (c++ function)": [[0, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", false]], "dequantize() (in module mlx.core)": [[126, "mlx.core.dequantize", false]], "device (class in mlx.core)": [[10, "mlx.core.Device", false]], "device_info() (in module mlx.core.metal)": [[238, "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)": [[127, "mlx.core.diag", false]], "diagonal (c++ function)": [[0, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", false]], "diagonal() (array method)": [[47, "mlx.core.array.diagonal", false]], "diagonal() (in module mlx.core)": [[128, "mlx.core.diagonal", false]], "disable_compile() (in module mlx.core)": [[129, "mlx.core.disable_compile", false]], "divide (c++ function)": [[0, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", false]], "divide() (in module mlx.core)": [[138, "mlx.core.divide", false]], "divmod (c++ function)": [[0, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", false]], "divmod() (in module mlx.core)": [[139, "mlx.core.divmod", false]], "dropout (class in mlx.nn)": [[369, "mlx.nn.Dropout", false]], "dropout2d (class in mlx.nn)": [[370, "mlx.nn.Dropout2d", false]], "dropout3d (class in mlx.nn)": [[371, "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]], "eig() (in module mlx.core.linalg)": [[207, "mlx.core.linalg.eig", false]], "eigh() (in module mlx.core.linalg)": [[208, "mlx.core.linalg.eigh", false]], "eigvals() (in module mlx.core.linalg)": [[209, "mlx.core.linalg.eigvals", false]], "eigvalsh() (in module mlx.core.linalg)": [[210, "mlx.core.linalg.eigvalsh", false]], "einsum() (in module mlx.core)": [[140, "mlx.core.einsum", false]], "einsum_path() (in module mlx.core)": [[141, "mlx.core.einsum_path", false]], "elu (class in mlx.nn)": [[372, "mlx.nn.ELU", false], [445, "mlx.nn.elu", false]], "embedding (class in mlx.nn)": [[373, "mlx.nn.Embedding", false]], "enable_compile() (in module mlx.core)": [[142, "mlx.core.enable_compile", false]], "equal (c++ function)": [[0, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", false]], "equal() (in module mlx.core)": [[143, "mlx.core.equal", false]], "erf (c++ function)": [[0, "_CPPv43erfRK5array14StreamOrDevice", false]], "erf() (in module mlx.core)": [[144, "mlx.core.erf", false]], "erfinv (c++ function)": [[0, "_CPPv46erfinvRK5array14StreamOrDevice", false]], "erfinv() (in module mlx.core)": [[145, "mlx.core.erfinv", false]], "eval() (in module mlx.core)": [[146, "mlx.core.eval", false]], "eval() (module method)": [[395, "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)": [[147, "mlx.core.exp", false]], "expand_dims (c++ function)": [[0, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", false]], "expand_dims() (in module mlx.core)": [[148, "mlx.core.expand_dims", false]], "expm1 (c++ function)": [[0, "_CPPv45expm1RK5array14StreamOrDevice", false]], "expm1() (in module mlx.core)": [[149, "mlx.core.expm1", false]], "exponential_decay() (in module mlx.optimizers)": [[507, "mlx.optimizers.exponential_decay", false]], "export_function() (in module mlx.core)": [[150, "mlx.core.export_function", false]], "export_to_dot() (in module mlx.core)": [[151, "mlx.core.export_to_dot", false]], "exporter() (in module mlx.core)": [[152, "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)": [[153, "mlx.core.eye", false]], "fft() (in module mlx.core.fft)": [[160, "mlx.core.fft.fft", false]], "fft2() (in module mlx.core.fft)": [[161, "mlx.core.fft.fft2", false]], "fftn() (in module mlx.core.fft)": [[162, "mlx.core.fft.fftn", false]], "fftshift() (in module mlx.core.fft)": [[163, "mlx.core.fft.fftshift", false]], "filter_and_map() (module method)": [[396, "mlx.nn.Module.filter_and_map", false]], "finfo (class in mlx.core)": [[174, "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)": [[175, "mlx.core.flatten", false]], "floor (c++ function)": [[0, "_CPPv45floorRK5array14StreamOrDevice", false]], "floor() (in module mlx.core)": [[176, "mlx.core.floor", false]], "floor_divide (c++ function)": [[0, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", false]], "floor_divide() (in module mlx.core)": [[177, "mlx.core.floor_divide", false]], "freeze() (module method)": [[397, "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)": [[178, "mlx.core.full", false]], "gather (c++ function)": [[0, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", false], [0, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", false]], "gather_mm (c++ function)": [[0, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", false]], "gather_mm() (in module mlx.core)": [[179, "mlx.core.gather_mm", false]], "gather_qmm (c++ function)": [[0, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", false]], "gather_qmm() (in module mlx.core)": [[180, "mlx.core.gather_qmm", false]], "gaussian_nll_loss (class in mlx.nn.losses)": [[459, "mlx.nn.losses.gaussian_nll_loss", false]], "gelu (class in mlx.nn)": [[374, "mlx.nn.GELU", false], [446, "mlx.nn.gelu", false]], "gelu_approx (class in mlx.nn)": [[447, "mlx.nn.gelu_approx", false]], "gelu_fast_approx (class in mlx.nn)": [[448, "mlx.nn.gelu_fast_approx", false]], "get_active_memory() (in module mlx.core)": [[181, "mlx.core.get_active_memory", false]], "get_cache_memory() (in module mlx.core)": [[182, "mlx.core.get_cache_memory", false]], "get_peak_memory() (in module mlx.core)": [[183, "mlx.core.get_peak_memory", false]], "glorot_normal() (in module mlx.nn.init)": [[437, "mlx.nn.init.glorot_normal", false]], "glorot_uniform() (in module mlx.nn.init)": [[438, "mlx.nn.init.glorot_uniform", false]], "glu (class in mlx.nn)": [[375, "mlx.nn.GLU", false], [449, "mlx.nn.glu", false]], "grad() (in module mlx.core)": [[184, "mlx.core.grad", false]], "greater (c++ function)": [[0, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", false]], "greater() (in module mlx.core)": [[185, "mlx.core.greater", false]], "greater_equal (c++ function)": [[0, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", false]], "greater_equal() (in module mlx.core)": [[186, "mlx.core.greater_equal", false]], "group (class in mlx.core.distributed)": [[130, "mlx.core.distributed.Group", false]], "groupnorm (class in mlx.nn)": [[377, "mlx.nn.GroupNorm", false]], "gru (class in mlx.nn)": [[376, "mlx.nn.GRU", false]], "gumbel() (in module mlx.core.random)": [[263, "mlx.core.random.gumbel", false]], "hadamard_transform (c++ function)": [[0, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", false]], "hadamard_transform() (in module mlx.core)": [[187, "mlx.core.hadamard_transform", false]], "hard_shrink (class in mlx.nn)": [[450, "mlx.nn.hard_shrink", false]], "hard_tanh (class in mlx.nn)": [[451, "mlx.nn.hard_tanh", false]], "hardshrink (class in mlx.nn)": [[378, "mlx.nn.HardShrink", false]], "hardswish (class in mlx.nn)": [[380, "mlx.nn.Hardswish", false], [452, "mlx.nn.hardswish", false]], "hardtanh (class in mlx.nn)": [[379, "mlx.nn.HardTanh", false]], "he_normal() (in module mlx.nn.init)": [[439, "mlx.nn.init.he_normal", false]], "he_uniform() (in module mlx.nn.init)": [[440, "mlx.nn.init.he_uniform", false]], "hinge_loss (class in mlx.nn.losses)": [[460, "mlx.nn.losses.hinge_loss", false]], "huber_loss (class in mlx.nn.losses)": [[461, "mlx.nn.losses.huber_loss", false]], "identity (c++ function)": [[0, "_CPPv48identityi14StreamOrDevice", false], [0, "_CPPv48identityi5Dtype14StreamOrDevice", false]], "identity() (in module mlx.core)": [[188, "mlx.core.identity", false]], "identity() (in module mlx.nn.init)": [[441, "mlx.nn.init.identity", false]], "ifft() (in module mlx.core.fft)": [[164, "mlx.core.fft.ifft", false]], "ifft2() (in module mlx.core.fft)": [[165, "mlx.core.fft.ifft2", false]], "ifftn() (in module mlx.core.fft)": [[166, "mlx.core.fft.ifftn", false]], "ifftshift() (in module mlx.core.fft)": [[167, "mlx.core.fft.ifftshift", false]], "imag (array property)": [[51, "mlx.core.array.imag", false]], "imag (c++ function)": [[0, "_CPPv44imagRK5array14StreamOrDevice", false]], "imag() (in module mlx.core)": [[189, "mlx.core.imag", false]], "import_function() (in module mlx.core)": [[190, "mlx.core.import_function", false]], "init() (in module mlx.core.distributed)": [[133, "mlx.core.distributed.init", false]], "init() (optimizer method)": [[501, "mlx.optimizers.Optimizer.init", false]], "inner (c++ function)": [[0, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", false]], "inner() (in module mlx.core)": [[191, "mlx.core.inner", false]], "instancenorm (class in mlx.nn)": [[381, "mlx.nn.InstanceNorm", false]], "inv() (in module mlx.core.linalg)": [[211, "mlx.core.linalg.inv", false]], "irfft() (in module mlx.core.fft)": [[168, "mlx.core.fft.irfft", false]], "irfft2() (in module mlx.core.fft)": [[169, "mlx.core.fft.irfft2", false]], "irfftn() (in module mlx.core.fft)": [[170, "mlx.core.fft.irfftn", false]], "is_available() (in module mlx.core.cuda)": [[117, "mlx.core.cuda.is_available", false]], "is_available() (in module mlx.core.distributed)": [[134, "mlx.core.distributed.is_available", false]], "is_available() (in module mlx.core.metal)": [[239, "mlx.core.metal.is_available", false]], "isclose (c++ function)": [[0, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "isclose() (in module mlx.core)": [[192, "mlx.core.isclose", false]], "isfinite (c++ function)": [[0, "_CPPv48isfiniteRK5array14StreamOrDevice", false]], "isfinite() (in module mlx.core)": [[193, "mlx.core.isfinite", false]], "isinf (c++ function)": [[0, "_CPPv45isinfRK5array14StreamOrDevice", false]], "isinf() (in module mlx.core)": [[194, "mlx.core.isinf", false]], "isnan (c++ function)": [[0, "_CPPv45isnanRK5array14StreamOrDevice", false]], "isnan() (in module mlx.core)": [[195, "mlx.core.isnan", false]], "isneginf (c++ function)": [[0, "_CPPv48isneginfRK5array14StreamOrDevice", false]], "isneginf() (in module mlx.core)": [[196, "mlx.core.isneginf", false]], "isposinf (c++ function)": [[0, "_CPPv48isposinfRK5array14StreamOrDevice", false]], "isposinf() (in module mlx.core)": [[197, "mlx.core.isposinf", false]], "issubdtype() (in module mlx.core)": [[198, "mlx.core.issubdtype", false]], "item() (array method)": [[52, "mlx.core.array.item", false]], "itemsize (array property)": [[53, "mlx.core.array.itemsize", false]], "join_schedules() (in module mlx.optimizers)": [[508, "mlx.optimizers.join_schedules", false]], "jvp() (in module mlx.core)": [[199, "mlx.core.jvp", false]], "key() (in module mlx.core.random)": [[264, "mlx.core.random.key", false]], "kl_div_loss (class in mlx.nn.losses)": [[462, "mlx.nn.losses.kl_div_loss", false]], "kron (c++ function)": [[0, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", false]], "kron() (in module mlx.core)": [[200, "mlx.core.kron", false]], "l1_loss (class in mlx.nn.losses)": [[463, "mlx.nn.losses.l1_loss", false]], "laplace() (in module mlx.core.random)": [[265, "mlx.core.random.laplace", false]], "layer_norm() (in module mlx.core.fast)": [[155, "mlx.core.fast.layer_norm", false]], "layernorm (class in mlx.nn)": [[383, "mlx.nn.LayerNorm", false]], "leaf_modules() (module method)": [[398, "mlx.nn.Module.leaf_modules", false]], "leaky_relu (class in mlx.nn)": [[453, "mlx.nn.leaky_relu", false]], "leakyrelu (class in mlx.nn)": [[384, "mlx.nn.LeakyReLU", false]], "left_shift (c++ function)": [[0, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", false]], "left_shift() (in module mlx.core)": [[201, "mlx.core.left_shift", false]], "less (c++ function)": [[0, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", false]], "less() (in module mlx.core)": [[202, "mlx.core.less", false]], "less_equal (c++ function)": [[0, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", false]], "less_equal() (in module mlx.core)": [[203, "mlx.core.less_equal", false]], "linear (class in mlx.nn)": [[385, "mlx.nn.Linear", false]], "linear_schedule() (in module mlx.optimizers)": [[509, "mlx.optimizers.linear_schedule", false]], "linspace (c++ function)": [[0, "_CPPv48linspaceddi5Dtype14StreamOrDevice", false]], "linspace() (in module mlx.core)": [[221, "mlx.core.linspace", false]], "lion (class in mlx.optimizers)": [[497, "mlx.optimizers.Lion", false]], "load() (in module mlx.core)": [[222, "mlx.core.load", false]], "load_weights() (module method)": [[399, "mlx.nn.Module.load_weights", false]], "log (c++ function)": [[0, "_CPPv43logRK5array14StreamOrDevice", false]], "log() (array method)": [[54, "mlx.core.array.log", false]], "log() (in module mlx.core)": [[223, "mlx.core.log", false]], "log10 (c++ function)": [[0, "_CPPv45log10RK5array14StreamOrDevice", false]], "log10() (array method)": [[55, "mlx.core.array.log10", false]], "log10() (in module mlx.core)": [[224, "mlx.core.log10", false]], "log1p (c++ function)": [[0, "_CPPv45log1pRK5array14StreamOrDevice", false]], "log1p() (array method)": [[56, "mlx.core.array.log1p", false]], "log1p() (in module mlx.core)": [[225, "mlx.core.log1p", false]], "log2 (c++ function)": [[0, "_CPPv44log2RK5array14StreamOrDevice", false]], "log2() (array method)": [[57, "mlx.core.array.log2", false]], "log2() (in module mlx.core)": [[226, "mlx.core.log2", false]], "log_cosh_loss (class in mlx.nn.losses)": [[464, "mlx.nn.losses.log_cosh_loss", false]], "log_sigmoid (class in mlx.nn)": [[454, "mlx.nn.log_sigmoid", false]], "log_softmax (class in mlx.nn)": [[455, "mlx.nn.log_softmax", false]], "logaddexp (c++ function)": [[0, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", false]], "logaddexp() (in module mlx.core)": [[227, "mlx.core.logaddexp", false]], "logcumsumexp (c++ function)": [[0, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", false]], "logcumsumexp() (array method)": [[58, "mlx.core.array.logcumsumexp", false]], "logcumsumexp() (in module mlx.core)": [[228, "mlx.core.logcumsumexp", false]], "logical_and (c++ function)": [[0, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", false]], "logical_and() (in module mlx.core)": [[229, "mlx.core.logical_and", false]], "logical_not (c++ function)": [[0, "_CPPv411logical_notRK5array14StreamOrDevice", false]], "logical_not() (in module mlx.core)": [[230, "mlx.core.logical_not", false]], "logical_or (c++ function)": [[0, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", false]], "logical_or() (in module mlx.core)": [[231, "mlx.core.logical_or", false]], "logsigmoid (class in mlx.nn)": [[386, "mlx.nn.LogSigmoid", false]], "logsoftmax (class in mlx.nn)": [[387, "mlx.nn.LogSoftmax", false]], "logsumexp (c++ function)": [[0, "_CPPv49logsumexpRK5array14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", false]], "logsumexp() (array method)": [[59, "mlx.core.array.logsumexp", false]], "logsumexp() (in module mlx.core)": [[232, "mlx.core.logsumexp", false]], "lstm (class in mlx.nn)": [[382, "mlx.nn.LSTM", false]], "lu() (in module mlx.core.linalg)": [[212, "mlx.core.linalg.lu", false]], "lu_factor() (in module mlx.core.linalg)": [[213, "mlx.core.linalg.lu_factor", false]], "margin_ranking_loss (class in mlx.nn.losses)": [[465, "mlx.nn.losses.margin_ranking_loss", false]], "matmul (c++ function)": [[0, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", false]], "matmul() (in module mlx.core)": [[233, "mlx.core.matmul", false]], "max (c++ function)": [[0, "_CPPv43maxRK5array14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayib14StreamOrDevice", false]], "max() (array method)": [[60, "mlx.core.array.max", false]], "max() (in module mlx.core)": [[234, "mlx.core.max", false]], "maximum (c++ function)": [[0, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", false]], "maximum() (in module mlx.core)": [[235, "mlx.core.maximum", false]], "maxpool1d (class in mlx.nn)": [[388, "mlx.nn.MaxPool1d", false]], "maxpool2d (class in mlx.nn)": [[389, "mlx.nn.MaxPool2d", false]], "maxpool3d (class in mlx.nn)": [[390, "mlx.nn.MaxPool3d", false]], "mean (c++ function)": [[0, "_CPPv44meanRK5array14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayib14StreamOrDevice", false]], "mean() (array method)": [[61, "mlx.core.array.mean", false]], "mean() (in module mlx.core)": [[236, "mlx.core.mean", false]], "meshgrid (c++ function)": [[0, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", false]], "meshgrid() (in module mlx.core)": [[237, "mlx.core.meshgrid", false]], "metal_kernel() (in module mlx.core.fast)": [[156, "mlx.core.fast.metal_kernel", false]], "min (c++ function)": [[0, "_CPPv43minRK5array14StreamOrDevice", false], [0, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayib14StreamOrDevice", false]], "min() (array method)": [[62, "mlx.core.array.min", false]], "min() (in module mlx.core)": [[242, "mlx.core.min", false]], "minimum (c++ function)": [[0, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", false]], "minimum() (in module mlx.core)": [[243, "mlx.core.minimum", false]], "mish (class in mlx.nn)": [[391, "mlx.nn.Mish", false], [470, "mlx.nn.mish", false]], "module (class in mlx.nn)": [[488, "mlx.nn.Module", false]], "modules() (module method)": [[400, "mlx.nn.Module.modules", false]], "moveaxis (c++ function)": [[0, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", false]], "moveaxis() (array method)": [[63, "mlx.core.array.moveaxis", false]], "moveaxis() (in module mlx.core)": [[244, "mlx.core.moveaxis", false]], "mse_loss (class in mlx.nn.losses)": [[466, "mlx.nn.losses.mse_loss", false]], "multiheadattention (class in mlx.nn)": [[412, "mlx.nn.MultiHeadAttention", false]], "multioptimizer (class in mlx.optimizers)": [[498, "mlx.optimizers.MultiOptimizer", false]], "multiply (c++ function)": [[0, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", false]], "multiply() (in module mlx.core)": [[245, "mlx.core.multiply", false]], "multivariate_normal() (in module mlx.core.random)": [[266, "mlx.core.random.multivariate_normal", false]], "muon (class in mlx.optimizers)": [[499, "mlx.optimizers.Muon", false]], "named_modules() (module method)": [[401, "mlx.nn.Module.named_modules", false]], "nan_to_num (c++ function)": [[0, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", false]], "nan_to_num() (in module mlx.core)": [[246, "mlx.core.nan_to_num", false]], "nbytes (array property)": [[64, "mlx.core.array.nbytes", false]], "ndim (array property)": [[65, "mlx.core.array.ndim", false]], "negative (c++ function)": [[0, "_CPPv48negativeRK5array14StreamOrDevice", false]], "negative() (in module mlx.core)": [[247, "mlx.core.negative", false]], "new_stream() (in module mlx.core)": [[248, "mlx.core.new_stream", false]], "nll_loss (class in mlx.nn.losses)": [[467, "mlx.nn.losses.nll_loss", false]], "norm() (in module mlx.core.linalg)": [[214, "mlx.core.linalg.norm", false]], "normal() (in module mlx.core.random)": [[267, "mlx.core.random.normal", false]], "normal() (in module mlx.nn.init)": [[442, "mlx.nn.init.normal", false]], "not_equal (c++ function)": [[0, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", false]], "not_equal() (in module mlx.core)": [[249, "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)": [[250, "mlx.core.ones", false]], "ones_like (c++ function)": [[0, "_CPPv49ones_likeRK5array14StreamOrDevice", false]], "ones_like() (in module mlx.core)": [[251, "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)": [[512, "mlx.optimizers.Optimizer", false]], "outer (c++ function)": [[0, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", false]], "outer() (in module mlx.core)": [[252, "mlx.core.outer", false]], "pad (c++ function)": [[0, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", false]], "pad() (in module mlx.core)": [[253, "mlx.core.pad", false]], "parameters() (module method)": [[402, "mlx.nn.Module.parameters", false]], "partition (c++ function)": [[0, "_CPPv49partitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv49partitionRK5arrayii14StreamOrDevice", false]], "partition() (in module mlx.core)": [[254, "mlx.core.partition", false]], "permutation() (in module mlx.core.random)": [[268, "mlx.core.random.permutation", false]], "pinv() (in module mlx.core.linalg)": [[215, "mlx.core.linalg.pinv", false]], "power (c++ function)": [[0, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", false]], "power() (in module mlx.core)": [[255, "mlx.core.power", false]], "prelu (class in mlx.nn)": [[413, "mlx.nn.PReLU", false], [471, "mlx.nn.prelu", false]], "prod (c++ function)": [[0, "_CPPv44prodRK5array14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayib14StreamOrDevice", false]], "prod() (array method)": [[66, "mlx.core.array.prod", false]], "prod() (in module mlx.core)": [[256, "mlx.core.prod", false]], "put_along_axis (c++ function)": [[0, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "put_along_axis() (in module mlx.core)": [[257, "mlx.core.put_along_axis", false]], "qr() (in module mlx.core.linalg)": [[216, "mlx.core.linalg.qr", false]], "quantize (c++ function)": [[0, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", false]], "quantize() (in module mlx.core)": [[258, "mlx.core.quantize", false]], "quantize() (in module mlx.nn)": [[336, "mlx.nn.quantize", false]], "quantized_matmul (c++ function)": [[0, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", false]], "quantized_matmul() (in module mlx.core)": [[259, "mlx.core.quantized_matmul", false]], "quantizedembedding (class in mlx.nn)": [[414, "mlx.nn.QuantizedEmbedding", false]], "quantizedlinear (class in mlx.nn)": [[415, "mlx.nn.QuantizedLinear", false]], "radians (c++ function)": [[0, "_CPPv47radiansRK5array14StreamOrDevice", false]], "radians() (in module mlx.core)": [[260, "mlx.core.radians", false]], "randint() (in module mlx.core.random)": [[269, "mlx.core.random.randint", false]], "real (array property)": [[67, "mlx.core.array.real", false]], "real (c++ function)": [[0, "_CPPv44realRK5array14StreamOrDevice", false]], "real() (in module mlx.core)": [[274, "mlx.core.real", false]], "reciprocal (c++ function)": [[0, "_CPPv410reciprocalRK5array14StreamOrDevice", false]], "reciprocal() (array method)": [[68, "mlx.core.array.reciprocal", false]], "reciprocal() (in module mlx.core)": [[275, "mlx.core.reciprocal", false]], "recv() (in module mlx.core.distributed)": [[135, "mlx.core.distributed.recv", false]], "recv_like() (in module mlx.core.distributed)": [[136, "mlx.core.distributed.recv_like", false]], "relu (class in mlx.nn)": [[418, "mlx.nn.ReLU", false], [472, "mlx.nn.relu", false]], "relu2 (class in mlx.nn)": [[419, "mlx.nn.ReLU2", false], [473, "mlx.nn.relu2", false]], "relu6 (class in mlx.nn)": [[420, "mlx.nn.ReLU6", false], [474, "mlx.nn.relu6", false]], "remainder (c++ function)": [[0, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", false]], "remainder() (in module mlx.core)": [[276, "mlx.core.remainder", false]], "repeat (c++ function)": [[0, "_CPPv46repeatRK5arrayi14StreamOrDevice", false], [0, "_CPPv46repeatRK5arrayii14StreamOrDevice", false]], "repeat() (in module mlx.core)": [[277, "mlx.core.repeat", false]], "reset_peak_memory() (in module mlx.core)": [[278, "mlx.core.reset_peak_memory", false]], "reshape (c++ function)": [[0, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", false]], "reshape() (array method)": [[69, "mlx.core.array.reshape", false]], "reshape() (in module mlx.core)": [[279, "mlx.core.reshape", false]], "rfft() (in module mlx.core.fft)": [[171, "mlx.core.fft.rfft", false]], "rfft2() (in module mlx.core.fft)": [[172, "mlx.core.fft.rfft2", false]], "rfftn() (in module mlx.core.fft)": [[173, "mlx.core.fft.rfftn", false]], "right_shift (c++ function)": [[0, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", false]], "right_shift() (in module mlx.core)": [[280, "mlx.core.right_shift", false]], "rms_norm() (in module mlx.core.fast)": [[157, "mlx.core.fast.rms_norm", false]], "rmsnorm (class in mlx.nn)": [[416, "mlx.nn.RMSNorm", false]], "rmsprop (class in mlx.optimizers)": [[504, "mlx.optimizers.RMSprop", false]], "rnn (class in mlx.nn)": [[417, "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)": [[281, "mlx.core.roll", false]], "rope (class in mlx.nn)": [[421, "mlx.nn.RoPE", false]], "rope() (in module mlx.core.fast)": [[158, "mlx.core.fast.rope", false]], "round (c++ function)": [[0, "_CPPv45roundRK5array14StreamOrDevice", false], [0, "_CPPv45roundRK5arrayi14StreamOrDevice", false]], "round() (array method)": [[70, "mlx.core.array.round", false]], "round() (in module mlx.core)": [[282, "mlx.core.round", false]], "rsqrt (c++ function)": [[0, "_CPPv45rsqrtRK5array14StreamOrDevice", false]], "rsqrt() (array method)": [[71, "mlx.core.array.rsqrt", false]], "rsqrt() (in module mlx.core)": [[283, "mlx.core.rsqrt", false]], "save() (in module mlx.core)": [[284, "mlx.core.save", false]], "save_gguf() (in module mlx.core)": [[285, "mlx.core.save_gguf", false]], "save_safetensors() (in module mlx.core)": [[286, "mlx.core.save_safetensors", false]], "save_weights() (module method)": [[403, "mlx.nn.Module.save_weights", false]], "savez() (in module mlx.core)": [[287, "mlx.core.savez", false]], "savez_compressed() (in module mlx.core)": [[288, "mlx.core.savez_compressed", false]], "scaled_dot_product_attention() (in module mlx.core.fast)": [[159, "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)": [[270, "mlx.core.random.seed", false]], "segmented_mm (c++ function)": [[0, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", false]], "selu (class in mlx.nn)": [[422, "mlx.nn.SELU", false], [475, "mlx.nn.selu", false]], "send() (in module mlx.core.distributed)": [[137, "mlx.core.distributed.send", false]], "sequential (class in mlx.nn)": [[423, "mlx.nn.Sequential", false]], "set_cache_limit() (in module mlx.core)": [[289, "mlx.core.set_cache_limit", false]], "set_default_device() (in module mlx.core)": [[290, "mlx.core.set_default_device", false]], "set_default_stream() (in module mlx.core)": [[291, "mlx.core.set_default_stream", false]], "set_dtype() (module method)": [[404, "mlx.nn.Module.set_dtype", false]], "set_memory_limit() (in module mlx.core)": [[292, "mlx.core.set_memory_limit", false]], "set_wired_limit() (in module mlx.core)": [[293, "mlx.core.set_wired_limit", false]], "sgd (class in mlx.optimizers)": [[505, "mlx.optimizers.SGD", false]], "shape (array property)": [[72, "mlx.core.array.shape", false]], "sigmoid (c++ function)": [[0, "_CPPv47sigmoidRK5array14StreamOrDevice", false]], "sigmoid (class in mlx.nn)": [[425, "mlx.nn.Sigmoid", false], [476, "mlx.nn.sigmoid", false]], "sigmoid() (in module mlx.core)": [[294, "mlx.core.sigmoid", false]], "sign (c++ function)": [[0, "_CPPv44signRK5array14StreamOrDevice", false]], "sign() (in module mlx.core)": [[295, "mlx.core.sign", false]], "silu (class in mlx.nn)": [[424, "mlx.nn.SiLU", false], [477, "mlx.nn.silu", false]], "sin (c++ function)": [[0, "_CPPv43sinRK5array14StreamOrDevice", false]], "sin() (array method)": [[73, "mlx.core.array.sin", false]], "sin() (in module mlx.core)": [[296, "mlx.core.sin", false]], "sinh (c++ function)": [[0, "_CPPv44sinhRK5array14StreamOrDevice", false]], "sinh() (in module mlx.core)": [[297, "mlx.core.sinh", false]], "sinusoidalpositionalencoding (class in mlx.nn)": [[426, "mlx.nn.SinusoidalPositionalEncoding", false]], "size (array property)": [[74, "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)": [[298, "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)": [[299, "mlx.core.slice_update", false]], "smooth_l1_loss (class in mlx.nn.losses)": [[468, "mlx.nn.losses.smooth_l1_loss", false]], "softmax (c++ function)": [[0, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayib14StreamOrDevice", false]], "softmax (class in mlx.nn)": [[427, "mlx.nn.Softmax", false], [478, "mlx.nn.softmax", false]], "softmax() (in module mlx.core)": [[300, "mlx.core.softmax", false]], "softmin (class in mlx.nn)": [[428, "mlx.nn.Softmin", false], [479, "mlx.nn.softmin", false]], "softplus (class in mlx.nn)": [[429, "mlx.nn.Softplus", false], [480, "mlx.nn.softplus", false]], "softshrink (class in mlx.nn)": [[430, "mlx.nn.Softshrink", false], [481, "mlx.nn.softshrink", false]], "softsign (class in mlx.nn)": [[431, "mlx.nn.Softsign", false]], "solve() (in module mlx.core.linalg)": [[217, "mlx.core.linalg.solve", false]], "solve_triangular() (in module mlx.core.linalg)": [[218, "mlx.core.linalg.solve_triangular", false]], "sort (c++ function)": [[0, "_CPPv44sortRK5array14StreamOrDevice", false], [0, "_CPPv44sortRK5arrayi14StreamOrDevice", false]], "sort() (in module mlx.core)": [[301, "mlx.core.sort", false]], "split (c++ function)": [[0, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayi14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayii14StreamOrDevice", false]], "split() (array method)": [[75, "mlx.core.array.split", false]], "split() (in module mlx.core)": [[302, "mlx.core.split", false]], "split() (in module mlx.core.random)": [[271, "mlx.core.random.split", false]], "sqrt (c++ function)": [[0, "_CPPv44sqrtRK5array14StreamOrDevice", false]], "sqrt() (array method)": [[76, "mlx.core.array.sqrt", false]], "sqrt() (in module mlx.core)": [[303, "mlx.core.sqrt", false]], "square (c++ function)": [[0, "_CPPv46squareRK5array14StreamOrDevice", false]], "square() (array method)": [[77, "mlx.core.array.square", false]], "square() (in module mlx.core)": [[304, "mlx.core.square", false]], "squeeze (c++ function)": [[0, "_CPPv47squeezeRK5array14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayi14StreamOrDevice", false]], "squeeze() (array method)": [[78, "mlx.core.array.squeeze", false]], "squeeze() (in module mlx.core)": [[305, "mlx.core.squeeze", false]], "stack (c++ function)": [[0, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", false]], "stack() (in module mlx.core)": [[306, "mlx.core.stack", false]], "start_capture() (in module mlx.core.metal)": [[240, "mlx.core.metal.start_capture", false]], "state (module property)": [[405, "mlx.nn.Module.state", false]], "state (optimizer property)": [[502, "mlx.optimizers.Optimizer.state", false]], "std (c++ function)": [[0, "_CPPv4StRK5array14StreamOrDevice", false], [0, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv4StRK5arraybi14StreamOrDevice", false], [0, "_CPPv4StRK5arrayibi14StreamOrDevice", false]], "std() (array method)": [[79, "mlx.core.array.std", false]], "std() (in module mlx.core)": [[307, "mlx.core.std", false]], "step (class in mlx.nn)": [[432, "mlx.nn.Step", false], [482, "mlx.nn.step", false]], "step_decay() (in module mlx.optimizers)": [[510, "mlx.optimizers.step_decay", false]], "stop_capture() (in module mlx.core.metal)": [[241, "mlx.core.metal.stop_capture", false]], "stop_gradient (c++ function)": [[0, "_CPPv413stop_gradientRK5array14StreamOrDevice", false]], "stop_gradient() (in module mlx.core)": [[308, "mlx.core.stop_gradient", false]], "stream (class in mlx.core)": [[344, "mlx.core.Stream", false]], "stream() (in module mlx.core)": [[309, "mlx.core.stream", false]], "subtract (c++ function)": [[0, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", false]], "subtract() (in module mlx.core)": [[310, "mlx.core.subtract", false]], "sum (c++ function)": [[0, "_CPPv43sumRK5array14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayib14StreamOrDevice", false]], "sum() (array method)": [[80, "mlx.core.array.sum", false]], "sum() (in module mlx.core)": [[311, "mlx.core.sum", false]], "svd() (in module mlx.core.linalg)": [[219, "mlx.core.linalg.svd", false]], "swapaxes (c++ function)": [[0, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", false]], "swapaxes() (array method)": [[81, "mlx.core.array.swapaxes", false]], "swapaxes() (in module mlx.core)": [[312, "mlx.core.swapaxes", false]], "synchronize() (in module mlx.core)": [[313, "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)": [[314, "mlx.core.take", false]], "take_along_axis (c++ function)": [[0, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", false]], "take_along_axis() (in module mlx.core)": [[315, "mlx.core.take_along_axis", false]], "tan (c++ function)": [[0, "_CPPv43tanRK5array14StreamOrDevice", false]], "tan() (in module mlx.core)": [[316, "mlx.core.tan", false]], "tanh (c++ function)": [[0, "_CPPv44tanhRK5array14StreamOrDevice", false]], "tanh (class in mlx.nn)": [[433, "mlx.nn.Tanh", false], [483, "mlx.nn.tanh", false]], "tanh() (in module mlx.core)": [[317, "mlx.core.tanh", false]], "tensordot (c++ function)": [[0, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", false], [0, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", false]], "tensordot() (in module mlx.core)": [[318, "mlx.core.tensordot", false]], "tile (c++ function)": [[0, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "tile() (in module mlx.core)": [[319, "mlx.core.tile", false]], "tolist() (array method)": [[82, "mlx.core.array.tolist", false]], "topk (c++ function)": [[0, "_CPPv44topkRK5arrayi14StreamOrDevice", false], [0, "_CPPv44topkRK5arrayii14StreamOrDevice", false]], "topk() (in module mlx.core)": [[320, "mlx.core.topk", false]], "trace (c++ function)": [[0, "_CPPv45traceRK5array14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", false]], "trace() (in module mlx.core)": [[321, "mlx.core.trace", false]], "train() (module method)": [[406, "mlx.nn.Module.train", false]], "trainable_parameters() (module method)": [[407, "mlx.nn.Module.trainable_parameters", false]], "training (module property)": [[408, "mlx.nn.Module.training", false]], "transformer (class in mlx.nn)": [[434, "mlx.nn.Transformer", false]], "transpose (c++ function)": [[0, "_CPPv49transposeRK5array14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "transpose() (array method)": [[83, "mlx.core.array.transpose", false]], "transpose() (in module mlx.core)": [[322, "mlx.core.transpose", false]], "tree_flatten() (in module mlx.utils)": [[339, "mlx.utils.tree_flatten", false]], "tree_map() (in module mlx.utils)": [[340, "mlx.utils.tree_map", false]], "tree_map_with_path() (in module mlx.utils)": [[341, "mlx.utils.tree_map_with_path", false]], "tree_reduce() (in module mlx.utils)": [[342, "mlx.utils.tree_reduce", false]], "tree_unflatten() (in module mlx.utils)": [[343, "mlx.utils.tree_unflatten", false]], "tri (c++ function)": [[0, "_CPPv43trii5Dtype14StreamOrDevice", false], [0, "_CPPv43triiii5Dtype14StreamOrDevice", false]], "tri() (in module mlx.core)": [[323, "mlx.core.tri", false]], "tri_inv() (in module mlx.core.linalg)": [[220, "mlx.core.linalg.tri_inv", false]], "tril (c++ function)": [[0, "_CPPv44tril5arrayi14StreamOrDevice", false]], "tril() (in module mlx.core)": [[324, "mlx.core.tril", false]], "triplet_loss (class in mlx.nn.losses)": [[469, "mlx.nn.losses.triplet_loss", false]], "triu (c++ function)": [[0, "_CPPv44triu5arrayi14StreamOrDevice", false]], "triu() (in module mlx.core)": [[325, "mlx.core.triu", false]], "truncated_normal() (in module mlx.core.random)": [[272, "mlx.core.random.truncated_normal", false]], "unflatten (c++ function)": [[0, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", false]], "unflatten() (in module mlx.core)": [[326, "mlx.core.unflatten", false]], "unfreeze() (module method)": [[409, "mlx.nn.Module.unfreeze", false]], "uniform() (in module mlx.core.random)": [[273, "mlx.core.random.uniform", false]], "uniform() (in module mlx.nn.init)": [[443, "mlx.nn.init.uniform", false]], "update() (module method)": [[410, "mlx.nn.Module.update", false]], "update() (optimizer method)": [[503, "mlx.optimizers.Optimizer.update", false]], "update_modules() (module method)": [[411, "mlx.nn.Module.update_modules", false]], "upsample (class in mlx.nn)": [[435, "mlx.nn.Upsample", false]], "value_and_grad() (in module mlx.core)": [[327, "mlx.core.value_and_grad", false]], "value_and_grad() (in module mlx.nn)": [[337, "mlx.nn.value_and_grad", false]], "var (c++ function)": [[0, "_CPPv43varRK5array14StreamOrDevice", false], [0, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv43varRK5arraybi14StreamOrDevice", false], [0, "_CPPv43varRK5arrayibi14StreamOrDevice", false]], "var() (array method)": [[84, "mlx.core.array.var", false]], "var() (in module mlx.core)": [[328, "mlx.core.var", false]], "view (c++ function)": [[0, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", false]], "view() (array method)": [[85, "mlx.core.array.view", false]], "view() (in module mlx.core)": [[329, "mlx.core.view", false]], "vjp() (in module mlx.core)": [[330, "mlx.core.vjp", false]], "vmap() (in module mlx.core)": [[331, "mlx.core.vmap", false]], "where (c++ function)": [[0, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", false]], "where() (in module mlx.core)": [[332, "mlx.core.where", false]], "zeros (c++ function)": [[0, "_CPPv45zerosRK5Shape14StreamOrDevice", false], [0, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", false]], "zeros() (in module mlx.core)": [[333, "mlx.core.zeros", false]], "zeros_like (c++ function)": [[0, "_CPPv410zeros_likeRK5array14StreamOrDevice", false]], "zeros_like() (in module mlx.core)": [[334, "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_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::dilation"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::groups"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::input"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::output_padding"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::padding"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::s"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::stride"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::weight"], [0, 0, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::dilation"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::groups"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::input"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::output_padding"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::padding"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::s"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::stride"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::weight"], [0, 0, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::dilation"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::groups"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::input"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::output_padding"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::padding"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::s"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::stride"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "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, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::biases"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::bits"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::group_size"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::mode"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::s"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::scales"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "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_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::a"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::b"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::lhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::rhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::s"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::sorted_indices"], [0, 0, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::biases"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::bits"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::group_size"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::lhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::mode"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::rhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::s"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::scales"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::sorted_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::transpose"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::w"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "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, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::a"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::axis"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::inclusive"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::reverse"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::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, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::bits"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::group_size"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::mode"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::s"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::w"], [0, 0, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::biases"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::bits"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::group_size"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::mode"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::s"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::scales"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::transpose"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::w"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "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, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm"], [0, 1, 1, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm::a"], [0, 1, 1, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm::b"], [0, 1, 1, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm::s"], [0, 1, 1, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm::segments"], [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"], [344, 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"], [86, 5, 1, "", "array_equal"], [87, 5, 1, "", "as_strided"], [88, 5, 1, "", "async_eval"], [89, 5, 1, "", "atleast_1d"], [90, 5, 1, "", "atleast_2d"], [91, 5, 1, "", "atleast_3d"], [92, 5, 1, "", "bitwise_and"], [93, 5, 1, "", "bitwise_invert"], [94, 5, 1, "", "bitwise_or"], [95, 5, 1, "", "bitwise_xor"], [96, 5, 1, "", "block_masked_mm"], [97, 5, 1, "", "broadcast_arrays"], [98, 5, 1, "", "broadcast_to"], [99, 5, 1, "", "ceil"], [100, 5, 1, "", "clear_cache"], [101, 5, 1, "", "clip"], [102, 5, 1, "", "compile"], [103, 5, 1, "", "concatenate"], [104, 5, 1, "", "conj"], [105, 5, 1, "", "conjugate"], [106, 5, 1, "", "contiguous"], [107, 5, 1, "", "conv1d"], [108, 5, 1, "", "conv2d"], [109, 5, 1, "", "conv3d"], [110, 5, 1, "", "conv_general"], [111, 5, 1, "", "conv_transpose1d"], [112, 5, 1, "", "conv_transpose2d"], [113, 5, 1, "", "conv_transpose3d"], [114, 5, 1, "", "convolve"], [115, 5, 1, "", "cos"], [116, 5, 1, "", "cosh"], [118, 5, 1, "", "cummax"], [119, 5, 1, "", "cummin"], [120, 5, 1, "", "cumprod"], [121, 5, 1, "", "cumsum"], [122, 3, 1, "", "custom_function"], [123, 5, 1, "", "default_device"], [124, 5, 1, "", "default_stream"], [125, 5, 1, "", "degrees"], [126, 5, 1, "", "dequantize"], [127, 5, 1, "", "diag"], [128, 5, 1, "", "diagonal"], [129, 5, 1, "", "disable_compile"], [138, 5, 1, "", "divide"], [139, 5, 1, "", "divmod"], [140, 5, 1, "", "einsum"], [141, 5, 1, "", "einsum_path"], [142, 5, 1, "", "enable_compile"], [143, 5, 1, "", "equal"], [144, 5, 1, "", "erf"], [145, 5, 1, "", "erfinv"], [146, 5, 1, "", "eval"], [147, 5, 1, "", "exp"], [148, 5, 1, "", "expand_dims"], [149, 5, 1, "", "expm1"], [150, 5, 1, "", "export_function"], [151, 5, 1, "", "export_to_dot"], [152, 5, 1, "", "exporter"], [153, 5, 1, "", "eye"], [174, 3, 1, "", "finfo"], [175, 5, 1, "", "flatten"], [176, 5, 1, "", "floor"], [177, 5, 1, "", "floor_divide"], [178, 5, 1, "", "full"], [179, 5, 1, "", "gather_mm"], [180, 5, 1, "", "gather_qmm"], [181, 5, 1, "", "get_active_memory"], [182, 5, 1, "", "get_cache_memory"], [183, 5, 1, "", "get_peak_memory"], [184, 5, 1, "", "grad"], [185, 5, 1, "", "greater"], [186, 5, 1, "", "greater_equal"], [187, 5, 1, "", "hadamard_transform"], [188, 5, 1, "", "identity"], [189, 5, 1, "", "imag"], [190, 5, 1, "", "import_function"], [191, 5, 1, "", "inner"], [192, 5, 1, "", "isclose"], [193, 5, 1, "", "isfinite"], [194, 5, 1, "", "isinf"], [195, 5, 1, "", "isnan"], [196, 5, 1, "", "isneginf"], [197, 5, 1, "", "isposinf"], [198, 5, 1, "", "issubdtype"], [199, 5, 1, "", "jvp"], [200, 5, 1, "", "kron"], [201, 5, 1, "", "left_shift"], [202, 5, 1, "", "less"], [203, 5, 1, "", "less_equal"], [221, 5, 1, "", "linspace"], [222, 5, 1, "", "load"], [223, 5, 1, "", "log"], [224, 5, 1, "", "log10"], [225, 5, 1, "", "log1p"], [226, 5, 1, "", "log2"], [227, 5, 1, "", "logaddexp"], [228, 5, 1, "", "logcumsumexp"], [229, 5, 1, "", "logical_and"], [230, 5, 1, "", "logical_not"], [231, 5, 1, "", "logical_or"], [232, 5, 1, "", "logsumexp"], [233, 5, 1, "", "matmul"], [234, 5, 1, "", "max"], [235, 5, 1, "", "maximum"], [236, 5, 1, "", "mean"], [237, 5, 1, "", "meshgrid"], [242, 5, 1, "", "min"], [243, 5, 1, "", "minimum"], [244, 5, 1, "", "moveaxis"], [245, 5, 1, "", "multiply"], [246, 5, 1, "", "nan_to_num"], [247, 5, 1, "", "negative"], [248, 5, 1, "", "new_stream"], [249, 5, 1, "", "not_equal"], [250, 5, 1, "", "ones"], [251, 5, 1, "", "ones_like"], [252, 5, 1, "", "outer"], [253, 5, 1, "", "pad"], [254, 5, 1, "", "partition"], [255, 5, 1, "", "power"], [256, 5, 1, "", "prod"], [257, 5, 1, "", "put_along_axis"], [258, 5, 1, "", "quantize"], [259, 5, 1, "", "quantized_matmul"], [260, 5, 1, "", "radians"], [274, 5, 1, "", "real"], [275, 5, 1, "", "reciprocal"], [276, 5, 1, "", "remainder"], [277, 5, 1, "", "repeat"], [278, 5, 1, "", "reset_peak_memory"], [279, 5, 1, "", "reshape"], [280, 5, 1, "", "right_shift"], [281, 5, 1, "", "roll"], [282, 5, 1, "", "round"], [283, 5, 1, "", "rsqrt"], [284, 5, 1, "", "save"], [285, 5, 1, "", "save_gguf"], [286, 5, 1, "", "save_safetensors"], [287, 5, 1, "", "savez"], [288, 5, 1, "", "savez_compressed"], [289, 5, 1, "", "set_cache_limit"], [290, 5, 1, "", "set_default_device"], [291, 5, 1, "", "set_default_stream"], [292, 5, 1, "", "set_memory_limit"], [293, 5, 1, "", "set_wired_limit"], [294, 5, 1, "", "sigmoid"], [295, 5, 1, "", "sign"], [296, 5, 1, "", "sin"], [297, 5, 1, "", "sinh"], [298, 5, 1, "", "slice"], [299, 5, 1, "", "slice_update"], [300, 5, 1, "", "softmax"], [301, 5, 1, "", "sort"], [302, 5, 1, "", "split"], [303, 5, 1, "", "sqrt"], [304, 5, 1, "", "square"], [305, 5, 1, "", "squeeze"], [306, 5, 1, "", "stack"], [307, 5, 1, "", "std"], [308, 5, 1, "", "stop_gradient"], [309, 5, 1, "", "stream"], [310, 5, 1, "", "subtract"], [311, 5, 1, "", "sum"], [312, 5, 1, "", "swapaxes"], [313, 5, 1, "", "synchronize"], [314, 5, 1, "", "take"], [315, 5, 1, "", "take_along_axis"], [316, 5, 1, "", "tan"], [317, 5, 1, "", "tanh"], [318, 5, 1, "", "tensordot"], [319, 5, 1, "", "tile"], [320, 5, 1, "", "topk"], [321, 5, 1, "", "trace"], [322, 5, 1, "", "transpose"], [323, 5, 1, "", "tri"], [324, 5, 1, "", "tril"], [325, 5, 1, "", "triu"], [326, 5, 1, "", "unflatten"], [327, 5, 1, "", "value_and_grad"], [328, 5, 1, "", "var"], [329, 5, 1, "", "view"], [330, 5, 1, "", "vjp"], [331, 5, 1, "", "vmap"], [332, 5, 1, "", "where"], [333, 5, 1, "", "zeros"], [334, 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": [[344, 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, 6, 1, "", "imag"], [52, 4, 1, "", "item"], [53, 6, 1, "", "itemsize"], [54, 4, 1, "", "log"], [55, 4, 1, "", "log10"], [56, 4, 1, "", "log1p"], [57, 4, 1, "", "log2"], [58, 4, 1, "", "logcumsumexp"], [59, 4, 1, "", "logsumexp"], [60, 4, 1, "", "max"], [61, 4, 1, "", "mean"], [62, 4, 1, "", "min"], [63, 4, 1, "", "moveaxis"], [64, 6, 1, "", "nbytes"], [65, 6, 1, "", "ndim"], [66, 4, 1, "", "prod"], [67, 6, 1, "", "real"], [68, 4, 1, "", "reciprocal"], [69, 4, 1, "", "reshape"], [70, 4, 1, "", "round"], [71, 4, 1, "", "rsqrt"], [72, 6, 1, "", "shape"], [73, 4, 1, "", "sin"], [74, 6, 1, "", "size"], [75, 4, 1, "", "split"], [76, 4, 1, "", "sqrt"], [77, 4, 1, "", "square"], [78, 4, 1, "", "squeeze"], [79, 4, 1, "", "std"], [80, 4, 1, "", "sum"], [81, 4, 1, "", "swapaxes"], [82, 4, 1, "", "tolist"], [83, 4, 1, "", "transpose"], [84, 4, 1, "", "var"], [85, 4, 1, "", "view"]], "mlx.core.cuda": [[117, 5, 1, "", "is_available"]], "mlx.core.custom_function": [[122, 4, 1, "", "__init__"]], "mlx.core.distributed": [[130, 3, 1, "", "Group"], [131, 5, 1, "", "all_gather"], [132, 5, 1, "", "all_sum"], [133, 5, 1, "", "init"], [134, 5, 1, "", "is_available"], [135, 5, 1, "", "recv"], [136, 5, 1, "", "recv_like"], [137, 5, 1, "", "send"]], "mlx.core.distributed.Group": [[130, 4, 1, "", "__init__"]], "mlx.core.fast": [[154, 5, 1, "", "cuda_kernel"], [155, 5, 1, "", "layer_norm"], [156, 5, 1, "", "metal_kernel"], [157, 5, 1, "", "rms_norm"], [158, 5, 1, "", "rope"], [159, 5, 1, "", "scaled_dot_product_attention"]], "mlx.core.fft": [[160, 5, 1, "", "fft"], [161, 5, 1, "", "fft2"], [162, 5, 1, "", "fftn"], [163, 5, 1, "", "fftshift"], [164, 5, 1, "", "ifft"], [165, 5, 1, "", "ifft2"], [166, 5, 1, "", "ifftn"], [167, 5, 1, "", "ifftshift"], [168, 5, 1, "", "irfft"], [169, 5, 1, "", "irfft2"], [170, 5, 1, "", "irfftn"], [171, 5, 1, "", "rfft"], [172, 5, 1, "", "rfft2"], [173, 5, 1, "", "rfftn"]], "mlx.core.finfo": [[174, 4, 1, "", "__init__"]], "mlx.core.linalg": [[204, 5, 1, "", "cholesky"], [205, 5, 1, "", "cholesky_inv"], [206, 5, 1, "", "cross"], [207, 5, 1, "", "eig"], [208, 5, 1, "", "eigh"], [209, 5, 1, "", "eigvals"], [210, 5, 1, "", "eigvalsh"], [211, 5, 1, "", "inv"], [212, 5, 1, "", "lu"], [213, 5, 1, "", "lu_factor"], [214, 5, 1, "", "norm"], [215, 5, 1, "", "pinv"], [216, 5, 1, "", "qr"], [217, 5, 1, "", "solve"], [218, 5, 1, "", "solve_triangular"], [219, 5, 1, "", "svd"], [220, 5, 1, "", "tri_inv"]], "mlx.core.metal": [[238, 5, 1, "", "device_info"], [239, 5, 1, "", "is_available"], [240, 5, 1, "", "start_capture"], [241, 5, 1, "", "stop_capture"]], "mlx.core.random": [[261, 5, 1, "", "bernoulli"], [262, 5, 1, "", "categorical"], [263, 5, 1, "", "gumbel"], [264, 5, 1, "", "key"], [265, 5, 1, "", "laplace"], [266, 5, 1, "", "multivariate_normal"], [267, 5, 1, "", "normal"], [268, 5, 1, "", "permutation"], [269, 5, 1, "", "randint"], [270, 5, 1, "", "seed"], [271, 5, 1, "", "split"], [272, 5, 1, "", "truncated_normal"], [273, 5, 1, "", "uniform"]], "mlx.nn": [[357, 3, 1, "", "ALiBi"], [358, 3, 1, "", "AvgPool1d"], [359, 3, 1, "", "AvgPool2d"], [360, 3, 1, "", "AvgPool3d"], [361, 3, 1, "", "BatchNorm"], [362, 3, 1, "", "CELU"], [363, 3, 1, "", "Conv1d"], [364, 3, 1, "", "Conv2d"], [365, 3, 1, "", "Conv3d"], [366, 3, 1, "", "ConvTranspose1d"], [367, 3, 1, "", "ConvTranspose2d"], [368, 3, 1, "", "ConvTranspose3d"], [369, 3, 1, "", "Dropout"], [370, 3, 1, "", "Dropout2d"], [371, 3, 1, "", "Dropout3d"], [372, 3, 1, "", "ELU"], [373, 3, 1, "", "Embedding"], [374, 3, 1, "", "GELU"], [375, 3, 1, "", "GLU"], [376, 3, 1, "", "GRU"], [377, 3, 1, "", "GroupNorm"], [378, 3, 1, "", "HardShrink"], [379, 3, 1, "", "HardTanh"], [380, 3, 1, "", "Hardswish"], [381, 3, 1, "", "InstanceNorm"], [382, 3, 1, "", "LSTM"], [383, 3, 1, "", "LayerNorm"], [384, 3, 1, "", "LeakyReLU"], [385, 3, 1, "", "Linear"], [386, 3, 1, "", "LogSigmoid"], [387, 3, 1, "", "LogSoftmax"], [388, 3, 1, "", "MaxPool1d"], [389, 3, 1, "", "MaxPool2d"], [390, 3, 1, "", "MaxPool3d"], [391, 3, 1, "", "Mish"], [488, 3, 1, "", "Module"], [412, 3, 1, "", "MultiHeadAttention"], [413, 3, 1, "", "PReLU"], [414, 3, 1, "", "QuantizedEmbedding"], [415, 3, 1, "", "QuantizedLinear"], [416, 3, 1, "", "RMSNorm"], [417, 3, 1, "", "RNN"], [418, 3, 1, "", "ReLU"], [419, 3, 1, "", "ReLU2"], [420, 3, 1, "", "ReLU6"], [421, 3, 1, "", "RoPE"], [422, 3, 1, "", "SELU"], [423, 3, 1, "", "Sequential"], [424, 3, 1, "", "SiLU"], [425, 3, 1, "", "Sigmoid"], [426, 3, 1, "", "SinusoidalPositionalEncoding"], [427, 3, 1, "", "Softmax"], [428, 3, 1, "", "Softmin"], [429, 3, 1, "", "Softplus"], [430, 3, 1, "", "Softshrink"], [431, 3, 1, "", "Softsign"], [432, 3, 1, "", "Step"], [433, 3, 1, "", "Tanh"], [434, 3, 1, "", "Transformer"], [435, 3, 1, "", "Upsample"], [335, 5, 1, "", "average_gradients"], [444, 3, 1, "", "celu"], [445, 3, 1, "", "elu"], [446, 3, 1, "", "gelu"], [447, 3, 1, "", "gelu_approx"], [448, 3, 1, "", "gelu_fast_approx"], [449, 3, 1, "", "glu"], [450, 3, 1, "", "hard_shrink"], [451, 3, 1, "", "hard_tanh"], [452, 3, 1, "", "hardswish"], [453, 3, 1, "", "leaky_relu"], [454, 3, 1, "", "log_sigmoid"], [455, 3, 1, "", "log_softmax"], [470, 3, 1, "", "mish"], [471, 3, 1, "", "prelu"], [336, 5, 1, "", "quantize"], [472, 3, 1, "", "relu"], [473, 3, 1, "", "relu2"], [474, 3, 1, "", "relu6"], [475, 3, 1, "", "selu"], [476, 3, 1, "", "sigmoid"], [477, 3, 1, "", "silu"], [478, 3, 1, "", "softmax"], [479, 3, 1, "", "softmin"], [480, 3, 1, "", "softplus"], [481, 3, 1, "", "softshrink"], [482, 3, 1, "", "step"], [483, 3, 1, "", "tanh"], [337, 5, 1, "", "value_and_grad"]], "mlx.nn.Module": [[392, 4, 1, "", "apply"], [393, 4, 1, "", "apply_to_modules"], [394, 4, 1, "", "children"], [395, 4, 1, "", "eval"], [396, 4, 1, "", "filter_and_map"], [397, 4, 1, "", "freeze"], [398, 4, 1, "", "leaf_modules"], [399, 4, 1, "", "load_weights"], [400, 4, 1, "", "modules"], [401, 4, 1, "", "named_modules"], [402, 4, 1, "", "parameters"], [403, 4, 1, "", "save_weights"], [404, 4, 1, "", "set_dtype"], [405, 6, 1, "", "state"], [406, 4, 1, "", "train"], [407, 4, 1, "", "trainable_parameters"], [408, 6, 1, "", "training"], [409, 4, 1, "", "unfreeze"], [410, 4, 1, "", "update"], [411, 4, 1, "", "update_modules"]], "mlx.nn.init": [[436, 5, 1, "", "constant"], [437, 5, 1, "", "glorot_normal"], [438, 5, 1, "", "glorot_uniform"], [439, 5, 1, "", "he_normal"], [440, 5, 1, "", "he_uniform"], [441, 5, 1, "", "identity"], [442, 5, 1, "", "normal"], [443, 5, 1, "", "uniform"]], "mlx.nn.losses": [[456, 3, 1, "", "binary_cross_entropy"], [457, 3, 1, "", "cosine_similarity_loss"], [458, 3, 1, "", "cross_entropy"], [459, 3, 1, "", "gaussian_nll_loss"], [460, 3, 1, "", "hinge_loss"], [461, 3, 1, "", "huber_loss"], [462, 3, 1, "", "kl_div_loss"], [463, 3, 1, "", "l1_loss"], [464, 3, 1, "", "log_cosh_loss"], [465, 3, 1, "", "margin_ranking_loss"], [466, 3, 1, "", "mse_loss"], [467, 3, 1, "", "nll_loss"], [468, 3, 1, "", "smooth_l1_loss"], [469, 3, 1, "", "triplet_loss"]], "mlx.optimizers": [[491, 3, 1, "", "AdaDelta"], [492, 3, 1, "", "Adafactor"], [493, 3, 1, "", "Adagrad"], [494, 3, 1, "", "Adam"], [495, 3, 1, "", "AdamW"], [496, 3, 1, "", "Adamax"], [497, 3, 1, "", "Lion"], [498, 3, 1, "", "MultiOptimizer"], [499, 3, 1, "", "Muon"], [512, 3, 1, "", "Optimizer"], [504, 3, 1, "", "RMSprop"], [505, 3, 1, "", "SGD"], [338, 5, 1, "", "clip_grad_norm"], [506, 5, 1, "", "cosine_decay"], [507, 5, 1, "", "exponential_decay"], [508, 5, 1, "", "join_schedules"], [509, 5, 1, "", "linear_schedule"], [510, 5, 1, "", "step_decay"]], "mlx.optimizers.Optimizer": [[500, 4, 1, "", "apply_gradients"], [501, 4, 1, "", "init"], [502, 6, 1, "", "state"], [503, 4, 1, "", "update"]], "mlx.utils": [[339, 5, 1, "", "tree_flatten"], [340, 5, 1, "", "tree_map"], [341, 5, 1, "", "tree_map_with_path"], [342, 5, 1, "", "tree_reduce"], [343, 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, 53, 65, 102, 124, 126, 161, 162, 165, 166, 169, 170, 172, 173, 184, 205, 214, 219, 222, 236, 252, 258, 262, 282, 285, 286, 307, 309, 327, 328, 329, 331, 337, 356, 359, 360, 376, 382, 389, 390, 396, 397, 399, 403, 404, 405, 409, 410, 411, 417, 490, 501, 502, 514, 517, 519, 520, 524, 525, 526, 527], "0": [0, 1, 2, 4, 5, 6, 7, 9, 10, 15, 19, 39, 46, 47, 50, 70, 75, 79, 84, 87, 88, 100, 103, 107, 108, 109, 110, 111, 112, 113, 122, 127, 128, 153, 154, 156, 159, 175, 179, 184, 190, 200, 207, 208, 211, 212, 214, 216, 220, 246, 253, 261, 265, 267, 268, 273, 277, 282, 289, 293, 298, 299, 302, 306, 307, 321, 323, 324, 325, 326, 327, 328, 331, 335, 338, 339, 341, 342, 356, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 377, 378, 381, 383, 384, 388, 389, 390, 413, 418, 421, 426, 430, 432, 434, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 447, 448, 450, 451, 452, 453, 456, 458, 460, 461, 465, 468, 469, 471, 472, 473, 474, 475, 481, 482, 485, 488, 491, 492, 494, 495, 496, 497, 499, 501, 504, 505, 506, 507, 508, 509, 510, 514, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526], "00005": 5, "0001": 426, "0005": 447, "001": 492, "00364": 5, "01": [5, 384, 453, 495, 499], "0137595": 439, "015": 448, "0184009": 440, "02264": 438, "024": 520, "02765": 439, "0300242": 440, "044715": [374, 447], "0485873": 458, "05": [17, 192, 361, 377, 381, 383, 416], "0507": 475, "05202": 6, "06": [459, 469, 491], "0638": 465, "06450": 383, "0645099": 442, "06561": 507, "06675": 497, "07467": 416, "08": [17, 192, 457, 493, 494, 495, 496, 504], "08022": 381, "081": 510, "08415": 448, "08494": 377, "08619": 440, "08681": [391, 470], "09864": 6, "0999938": 508, "0999961": 506, "0d": 499, "0f": 0, "0j": [207, 209], "1": [0, 1, 2, 3, 4, 6, 7, 9, 15, 19, 29, 30, 39, 47, 50, 88, 107, 108, 109, 110, 111, 112, 113, 122, 127, 128, 149, 150, 151, 152, 154, 156, 159, 160, 161, 164, 165, 168, 169, 170, 171, 172, 173, 175, 187, 191, 198, 200, 205, 206, 207, 208, 209, 210, 212, 214, 216, 233, 237, 252, 254, 258, 262, 265, 266, 267, 273, 292, 294, 298, 299, 301, 314, 320, 321, 326, 327, 338, 339, 341, 342, 347, 356, 358, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 375, 376, 377, 381, 382, 383, 385, 388, 413, 416, 417, 421, 425, 426, 432, 435, 437, 438, 439, 440, 441, 442, 443, 444, 445, 447, 448, 449, 451, 454, 455, 456, 457, 458, 459, 460, 461, 462, 464, 465, 467, 468, 469, 475, 476, 478, 479, 480, 482, 485, 488, 490, 491, 492, 493, 494, 495, 496, 497, 501, 504, 505, 506, 507, 508, 509, 510, 517, 518, 519, 520, 521, 522, 524, 525, 526, 527], "10": [0, 3, 6, 7, 200, 224, 282, 287, 340, 356, 399, 485, 508, 510, 517, 518, 521], "100": [2, 5, 6, 456, 509, 517, 520, 523, 527], "1000": [2, 159, 506, 517], "10000": 421, "101": 509, "1024": [1, 6], "105361": 456, "10_000": 5, "10x": 497, "11": 214, "12": [6, 9, 187, 200, 508], "1212": 491, "123": [518, 522], "12451": 438, "128": [159, 287, 356], "13": 9, "14": [9, 200], "15": [1, 9, 200, 214, 293, 342, 517], "150594": 437, "15268": 439, "16": [1, 154, 156, 347, 358, 360, 381, 388, 390, 392, 488], "1606": 448, "1607": [381, 383], "16384": 187, "16506": 440, "168": 518, "17": [4, 9], "177208": 439, "18": 200, "1803": 377, "1908": [391, 470], "1910": 416, "191107": 437, "192": 518, "1985": 214, "1_000": 5, "1_all": 9, "1d": [0, 107, 111, 114, 285, 315, 499], "1e": [0, 5, 7, 17, 192, 361, 377, 381, 383, 384, 416, 457, 459, 469, 490, 491, 492, 493, 494, 495, 496, 501, 504, 506, 507, 508, 509, 510], "1e3": 517, "1st": 258, "2": [0, 1, 2, 4, 5, 6, 7, 9, 39, 108, 112, 122, 127, 128, 144, 150, 151, 152, 159, 161, 165, 168, 169, 170, 171, 172, 173, 175, 187, 198, 200, 204, 205, 206, 207, 208, 209, 210, 211, 212, 214, 215, 216, 219, 220, 226, 233, 258, 266, 267, 271, 298, 299, 318, 321, 323, 324, 325, 326, 338, 342, 347, 356, 358, 359, 360, 364, 367, 374, 384, 388, 389, 390, 416, 426, 435, 436, 437, 438, 439, 440, 441, 442, 443, 447, 458, 459, 461, 468, 469, 473, 485, 488, 490, 491, 493, 494, 495, 501, 504, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "20": [187, 200, 214], "200": [6, 508, 520], "2002": 6, "2011": 493, "2012": [491, 504], "2015": [370, 494, 496], "2019": [6, 495], "2020": 6, "2021": 6, "20397": 456, "20_000": 6, "21": [6, 200, 510], "2104": 6, "223144": 456, "223404": 438, "225": 214, "225763": 465, "2302": 497, "23607": [214, 216], "24": 200, "24264": 214, "247": 6, "25": [9, 413, 435], "25211": 439, "256": [1, 7, 154, 156], "256995": 465, "27": 4, "28": [187, 200], "2d": [0, 108, 112, 128, 258, 361, 370], "2nd": 258, "2x": 524, "3": [0, 1, 2, 4, 6, 9, 88, 109, 113, 122, 150, 152, 175, 198, 200, 206, 207, 208, 209, 210, 214, 216, 298, 299, 326, 338, 342, 360, 365, 368, 374, 390, 435, 438, 440, 447, 452, 492, 497, 514, 517, 518, 519, 521, 524, 525], "30": 492, "3118": 524, "32": [1, 6, 7, 96, 159, 258, 259, 347, 359, 360, 389, 390, 416, 517], "32mib": 335, "330": 6, "33333": 435, "33554432": 335, "348587": 458, "35": 9, "363207": 437, "36788": 517, "379159": 438, "380709": 442, "39": 6, "3d": [0, 2, 109, 113, 158, 361, 371, 435], "3f": [2, 7, 517], "3x": 2, "4": [0, 1, 2, 6, 126, 156, 159, 175, 180, 200, 214, 258, 259, 287, 298, 326, 336, 342, 347, 358, 359, 360, 361, 381, 388, 389, 390, 414, 415, 434, 435, 437, 438, 439, 456, 517, 518, 519, 521, 525, 527], "4096": [2, 517, 520, 527], "40x": 1, "41421": 214, "417497": 443, "42": 343, "437": 6, "44": 6, "447214": 216, "458835": 439, "475": 6, "48095": 437, "4d": [1, 435, 499], "4m": 1, "5": [0, 1, 2, 5, 6, 9, 159, 200, 214, 261, 292, 298, 342, 358, 361, 369, 370, 371, 374, 378, 381, 388, 430, 435, 436, 439, 440, 447, 450, 468, 481, 485, 490, 499, 504, 506, 507, 517, 520, 521], "50": [0, 221], "500": [6, 527], "510826": 456, "512": [3, 6, 434, 527], "534422": 442, "539245": 456, "53947": 437, "54": 9, "55": 1, "550": 9, "559": 2, "5701": 491, "573409": 465, "57771": 216, "579": 6, "5f": 5, "6": [1, 2, 6, 122, 200, 214, 287, 298, 420, 434, 438, 447, 448, 452, 459, 469, 474, 504, 517, 521, 525], "61278": 437, "617261": 443, "628": 6, "633": 6, "639": 520, "64": [0, 1, 96, 126, 180, 258, 259, 336, 347, 414, 415], "64331": 440, "666329": 440, "66667": 435, "67326": 475, "676": 1, "690": 6, "6967": 439, "7": [2, 6, 9, 200, 214, 258, 521], "702": [374, 448], "707107": [207, 208], "71828": 517, "74166": 214, "74597": 214, "75": 435, "75596": 465, "75787": 439, "765166": 465, "773433": 465, "774": 2, "776856": 438, "793615": 440, "79854": 440, "7b": 6, "7m": 1, "8": [0, 1, 2, 6, 214, 258, 347, 359, 360, 381, 389, 390, 434, 457, 491, 492, 493, 494, 495, 496, 504, 517, 521, 525, 527], "8192": [6, 187], "84804": 214, "863726": 443, "883935": 443, "890597": 438, "894427": 216, "89613": 437, "8gb": 6, "8x": 1, "9": [4, 9, 214, 458, 491, 494, 495, 496, 497, 501, 507, 510, 524], "90041": 438, "912766": 438, "916291": 456, "95": [7, 499], "982273": 442, "99": [497, 504], "995016": 437, "999": [494, 495, 496], "A": [0, 2, 6, 8, 9, 10, 72, 86, 102, 150, 151, 154, 155, 156, 157, 159, 184, 198, 199, 205, 207, 208, 209, 210, 212, 214, 216, 219, 222, 232, 233, 234, 238, 242, 258, 261, 262, 263, 265, 266, 267, 268, 269, 272, 273, 302, 306, 309, 327, 330, 331, 336, 337, 338, 339, 340, 341, 342, 343, 344, 356, 361, 370, 376, 377, 381, 383, 396, 400, 401, 404, 410, 411, 416, 423, 426, 434, 437, 438, 440, 448, 469, 470, 488, 490, 494, 496, 498, 500, 501, 503, 508, 517, 518, 519, 520, 522, 523, 524], "AS": 179, "And": [4, 6, 435], "As": [7, 39, 314, 356, 518], "At": [101, 326, 518], "But": [519, 527], "By": [6, 336, 404, 456, 518, 520, 524], "For": [0, 1, 2, 4, 6, 9, 39, 122, 126, 159, 179, 198, 214, 258, 343, 356, 361, 370, 374, 392, 397, 406, 409, 415, 421, 426, 435, 437, 438, 439, 440, 456, 485, 490, 499, 514, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "If": [0, 1, 2, 4, 6, 9, 16, 17, 18, 19, 27, 28, 29, 30, 82, 86, 87, 88, 101, 103, 114, 118, 119, 120, 121, 127, 128, 131, 132, 133, 135, 136, 137, 146, 155, 157, 158, 159, 163, 167, 171, 172, 173, 177, 178, 179, 184, 192, 204, 205, 206, 214, 219, 222, 228, 232, 233, 234, 236, 237, 242, 246, 250, 253, 254, 256, 257, 262, 266, 267, 268, 277, 281, 289, 292, 300, 301, 302, 307, 311, 313, 314, 315, 318, 320, 321, 327, 328, 331, 333, 335, 336, 339, 340, 342, 361, 363, 364, 365, 366, 367, 368, 377, 383, 385, 397, 399, 409, 410, 411, 415, 417, 421, 423, 426, 435, 456, 458, 469, 492, 494, 495, 517, 518, 519, 520, 522, 523, 526, 527, 528], "In": [0, 1, 2, 6, 7, 39, 159, 233, 258, 340, 356, 370, 377, 488, 491, 493, 494, 496, 497, 500, 516, 517, 518, 519, 520, 522, 523, 526, 527], "It": [2, 6, 9, 136, 184, 291, 327, 338, 342, 356, 411, 415, 500, 512, 518, 522, 524, 526], "Its": [356, 519], "No": [2, 6, 208, 210, 518], "Not": [102, 249, 517], "ON": [3, 4, 9], "Of": 520, "On": [1, 517, 520, 523], "One": [160, 164, 171, 253, 283, 517, 519, 520, 522], "THE": 9, "That": [6, 267], "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, 51, 53, 64, 65, 67, 72, 82, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 125, 126, 127, 128, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 143, 144, 145, 147, 148, 149, 150, 151, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 177, 178, 179, 180, 182, 183, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 199, 200, 201, 202, 203, 206, 207, 208, 209, 210, 212, 213, 214, 216, 217, 218, 219, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 240, 242, 243, 244, 245, 247, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 285, 286, 289, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 347, 349, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 369, 370, 371, 373, 375, 376, 377, 381, 382, 383, 385, 388, 389, 390, 392, 393, 397, 399, 403, 404, 405, 406, 409, 410, 411, 412, 414, 415, 416, 417, 421, 423, 426, 432, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 482, 485, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 502, 504, 505, 506, 509, 512, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528], "Then": [5, 9], "There": [1, 356, 435, 517], "These": [1, 2, 102, 257, 315, 458, 527], "To": [0, 1, 2, 3, 5, 6, 7, 9, 212, 258, 289, 356, 485, 490, 517, 518, 519, 520, 525], "With": 519, "_": [1, 3, 5, 6, 341, 356, 506, 507, 508, 509, 510, 514, 517, 523, 527], "__call__": [1, 6, 7, 356, 488, 519], "__init__": [2, 6, 7, 10, 11, 12, 31, 122, 130, 174, 344, 356, 488], "__main__": [2, 6], "__name__": [2, 6], "_a": 2, "_ext": 2, "_f": 214, "_in": [437, 438], "_out": [437, 438], "_p": 469, "_val": 451, "a1": 179, "a2": 179, "a_": 214, "a_max": [0, 101], "a_min": [0, 101], "a_ndim": 1, "a_shap": 1, "a_strid": 1, "a_view": 524, "ab": [0, 17, 192, 214, 327, 377, 381, 383, 391, 416, 448, 470, 517, 519], "abil": 518, "abl": [2, 4, 522], "abort": 122, "about": [1, 2, 6, 7, 141, 238, 523, 527], "abov": [1, 2, 6, 258, 324, 356, 435, 518, 519, 520, 521, 522, 523, 527], "absolut": [0, 13, 17, 192, 447, 448, 468, 518], "acc": 342, "acceler": [4, 361], "accept": [518, 522], "access": [0, 6, 52, 356, 488, 501, 518, 523, 527], "accord": [0, 263, 332, 336, 412, 437, 438, 439, 440], "accordingli": 2, "accumul": [342, 416], "accuraci": 7, "accustom": 6, "achiev": [356, 518], "across": [1, 2, 9, 335, 377, 518], "act": [2, 464], "action": 356, "activ": [2, 9, 181, 370, 419, 432, 434, 450, 470, 473, 481, 482, 484, 517], "actual": [6, 19, 399, 488, 523], "ad": [0, 1, 2, 5, 9, 155, 341, 366, 367, 368, 381, 488, 491, 492, 493, 494, 495, 496, 504, 518, 523, 526], "adadelta": 490, "adafactor": 490, "adagrad": 490, "adam": [490, 496, 497, 508, 509], "adamax": 490, "adamw": [490, 497, 499], "adapt": [491, 492, 493, 518], "add": [0, 1, 2, 3, 4, 6, 15, 39, 148, 227, 253, 363, 364, 365, 366, 367, 368, 519, 520, 522, 527], "add_argu": 6, "add_depend": 2, "add_execut": 4, "add_fun": 519, "add_librari": 2, "addit": [0, 2, 4, 6, 9, 14, 15, 150, 155, 157, 159, 222, 361, 366, 367, 368, 377, 383, 412, 416, 488, 520], "addmm": 0, "address": 2, "adjac": 370, "advanc": [6, 517], "advantag": 527, "advis": 524, "affin": [0, 126, 180, 258, 259, 336, 361, 377, 381, 383, 385, 414, 415], "after": [2, 6, 7, 29, 100, 175, 177, 180, 254, 258, 361, 377, 383, 392, 393, 397, 399, 406, 409, 410, 411, 412, 434, 468, 517, 518, 527], "after_1": 253, "after_2": 253, "after_i": 253, "after_n": 253, "afternoon": 6, "again": [6, 9, 356, 517], "against": [0, 4, 97], "aggreg": [412, 518], "ago": 6, "ai": 122, "aim": 518, "ainv": [211, 220], "albeit": 527, "algebra": 8, "algorithm": [435, 497], "alia": [104, 105, 374], "alibi": 356, "align": [205, 258, 376, 382], "align_corn": 435, "all": [0, 1, 2, 3, 7, 9, 17, 29, 39, 89, 90, 91, 102, 108, 109, 110, 112, 113, 122, 131, 132, 133, 151, 153, 162, 163, 166, 167, 170, 173, 179, 180, 207, 209, 219, 233, 253, 254, 281, 305, 335, 336, 356, 392, 393, 397, 400, 401, 402, 407, 409, 412, 426, 434, 435, 485, 488, 512, 514, 517, 521, 522, 523, 525, 528], "all_avg": 518, "all_reduce_grad": 518, "all_reduce_s": 335, "all_sum": 518, "allclos": [0, 1, 154, 156], "alloc": [2, 182, 289, 292, 488], "allow": [0, 1, 2, 150, 152, 179, 180, 198, 338, 356, 411, 488, 512, 518, 521, 522, 525], "allow_col_major": [0, 106], "almost": [6, 518], "alon": [2, 524], "along": [0, 2, 27, 28, 102, 103, 118, 119, 120, 121, 131, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 179, 180, 187, 206, 214, 228, 257, 268, 277, 281, 300, 302, 306, 314, 315, 318, 319, 320, 321, 329, 356, 375, 417, 449], "alpha": [0, 2, 15, 258, 362, 372, 444, 445, 469, 471, 475, 495, 504], "alpha_": 2, "alreadi": [2, 3, 6, 518], "also": [0, 1, 2, 4, 6, 7, 8, 9, 12, 14, 92, 94, 95, 129, 138, 139, 143, 162, 166, 170, 173, 185, 186, 201, 202, 203, 227, 235, 243, 245, 249, 255, 258, 276, 280, 310, 336, 337, 349, 356, 396, 410, 412, 414, 415, 424, 446, 475, 477, 484, 490, 517, 518, 519, 520, 521, 522, 523, 524, 525, 528], "altern": 514, "although": 518, "alwai": [1, 87, 181, 190, 207, 209, 339, 518, 519, 520], "am": 6, "among": 2, "amount": [6, 183, 292, 358, 388, 519], "amus": 6, "an": [0, 1, 2, 3, 4, 6, 7, 9, 11, 16, 18, 31, 88, 89, 90, 91, 98, 106, 107, 108, 109, 110, 111, 112, 113, 130, 135, 136, 137, 146, 150, 152, 153, 155, 158, 159, 175, 178, 188, 190, 193, 204, 214, 222, 244, 250, 251, 253, 256, 257, 258, 259, 268, 277, 279, 281, 282, 292, 293, 302, 305, 312, 314, 315, 318, 319, 323, 326, 331, 333, 334, 339, 340, 341, 342, 347, 356, 369, 374, 377, 382, 383, 385, 392, 412, 413, 415, 417, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 447, 471, 485, 490, 491, 499, 503, 507, 512, 514, 516, 517, 518, 519, 520, 521, 523, 524, 525, 526, 527, 528], "anaconda": 518, "anchor": 469, "angl": [125, 260, 384], "angular": [158, 421], "ani": [0, 1, 2, 6, 8, 19, 102, 122, 133, 335, 339, 340, 341, 342, 343, 356, 374, 392, 393, 396, 405, 415, 434, 435, 485, 499, 509, 516, 517, 518, 520, 523, 525, 526, 527], "anonym": 517, "anoth": [0, 97, 101, 198, 233, 310, 332, 347, 356, 392, 517, 519, 520, 521, 527], "anwywher": 9, "anyhow": 6, "anymor": 6, "anyth": [6, 327, 518, 523], "anytim": 523, "api": [1, 2, 88, 150, 152, 190, 374, 518, 519, 520], "aplu": 215, "app": 9, "append": [6, 233, 517, 523], "appl": [2, 6, 8, 9, 527], "appli": [0, 39, 158, 159, 179, 219, 340, 341, 342, 356, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 370, 371, 372, 374, 375, 377, 378, 379, 380, 381, 383, 384, 385, 386, 387, 388, 389, 390, 391, 393, 406, 413, 415, 416, 417, 418, 419, 420, 422, 424, 425, 427, 428, 429, 430, 431, 432, 433, 435, 444, 445, 446, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 485, 494, 495, 500, 503, 509, 512, 517, 518], "applic": [3, 9], "apply_fn": 393, "apply_gradi": 490, "apply_to_modul": [356, 397], "approach": [464, 520], "appropri": 517, "approx": 374, "approxim": [17, 374, 446, 447, 448], "apt": 9, "ar": [0, 1, 2, 5, 6, 7, 8, 9, 17, 19, 86, 88, 96, 97, 98, 101, 102, 110, 114, 122, 126, 128, 133, 135, 136, 146, 153, 154, 156, 159, 161, 162, 165, 166, 169, 170, 172, 173, 175, 179, 180, 184, 192, 193, 194, 195, 196, 197, 198, 199, 207, 208, 209, 210, 212, 214, 216, 222, 233, 252, 253, 254, 258, 259, 261, 262, 263, 267, 268, 269, 272, 273, 281, 287, 288, 305, 306, 314, 327, 330, 331, 336, 339, 340, 347, 361, 363, 364, 365, 366, 367, 368, 369, 370, 371, 377, 381, 383, 385, 399, 412, 415, 435, 456, 458, 459, 484, 488, 490, 497, 501, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "arang": [0, 1, 214, 268, 347, 435, 521, 524], "arbitrari": [339, 488, 518], "arbitrarili": [1, 102, 356, 516, 520, 525], "arc": 0, "arcco": 0, "arccosh": 0, "architectur": [6, 9, 238, 356, 411, 527], "archiv": 526, "arcsin": 0, "arcsinh": 0, "arctan": 0, "arctan2": 0, "arctanh": 0, "arg": [2, 6, 11, 19, 88, 130, 146, 150, 151, 152, 174, 190, 287, 288, 344, 522], "arg1": 198, "arg2": 198, "argmax": [0, 7], "argmin": 0, "argnam": [184, 327], "argnum": [2, 122, 184, 327, 520], "argpars": 6, "argpartit": 0, "argsort": 0, "argument": [1, 32, 69, 83, 88, 102, 146, 184, 327, 340, 341, 342, 356, 435, 514, 518, 519, 520, 522, 526, 527, 528], "argumentpars": 6, "ari": [89, 90, 91], "aris": 524, "arm": 9, "arm64": 9, "around": 6, "arr": [0, 284, 521], "arr_0": 526, "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, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 122, 125, 126, 127, 128, 131, 132, 135, 136, 137, 138, 139, 140, 141, 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, 175, 176, 177, 178, 179, 180, 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, 242, 243, 244, 245, 246, 247, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 338, 347, 356, 361, 382, 392, 399, 402, 407, 413, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 482, 485, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 504, 505, 506, 507, 508, 509, 510, 517, 518, 519, 520, 523, 524, 525, 527], "array_equ": [0, 17, 192], "arrayfir": 8, "arxiv": [6, 377, 381, 383, 391, 416, 448, 470, 491, 497], "as_strid": 0, "ascend": [208, 210], "ask": [6, 518, 522], "assert": [1, 2, 154, 156], "assign": [0, 2, 39, 488, 518], "associ": [2, 287, 288, 523], "assum": [0, 6, 96, 206, 208, 210, 216, 340, 356, 377, 518], "ast": 1, "astyp": [0, 1, 2, 6, 154, 156, 392, 524], "asynchron": 88, "atleast": 0, "atleast_1d": 0, "atleast_2d": 0, "atleast_3d": 0, "atol": [0, 17, 192], "atom": [1, 156], "atomic_fetch_add_explicit": 1, "atomic_output": [1, 156], "attach": 2, "attempt": [102, 518], "attend": 412, "attent": [159, 397, 412, 426, 434], "attention_norm": 6, "attribut": [1, 10, 11, 12, 31, 174, 344, 405, 488, 512], "audio": 435, "auto": [0, 2, 4, 9, 154, 518, 519], "autom": 520, "automat": [1, 2, 8, 154, 156, 222, 518, 525, 526, 527], "autoregress": 6, "avail": [2, 5, 6, 7, 9, 11, 117, 133, 134, 239, 292, 349, 518, 522, 527], "averag": [335, 358, 359, 360, 491, 492, 494, 495, 496, 518], "avgpool1d": 356, "avgpool2d": 356, "avgpool3d": 356, "avoid": [1, 2, 404, 517, 518], "awai": [2, 6], "awar": [517, 523], "ax": [0, 2, 16, 18, 27, 28, 83, 122, 148, 161, 162, 163, 165, 166, 167, 169, 170, 172, 173, 175, 191, 214, 217, 218, 232, 234, 236, 242, 253, 256, 281, 298, 299, 300, 305, 307, 311, 312, 318, 322, 328, 520], "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, 58, 59, 60, 61, 62, 66, 75, 78, 79, 80, 84, 103, 118, 119, 120, 121, 128, 131, 148, 155, 157, 160, 164, 168, 169, 170, 171, 172, 173, 175, 187, 206, 212, 214, 228, 232, 234, 236, 242, 244, 253, 254, 256, 257, 262, 268, 277, 281, 300, 301, 302, 305, 306, 307, 311, 312, 314, 315, 319, 320, 321, 322, 326, 328, 329, 331, 358, 359, 360, 375, 388, 389, 390, 417, 449, 455, 457, 458, 462, 467, 469, 478, 479, 521], "axis1": [0, 47, 81, 128, 312, 321], "axis2": [0, 47, 81, 128, 312, 321], "axpbi": 2, "axpby_gener": 2, "axpby_general_": 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, 86, 92, 94, 95, 96, 138, 139, 143, 154, 156, 158, 159, 177, 179, 185, 186, 190, 191, 192, 200, 201, 202, 203, 206, 214, 217, 218, 227, 229, 231, 233, 235, 243, 245, 249, 252, 255, 258, 265, 276, 280, 310, 318, 327, 339, 341, 342, 375, 385, 417, 435, 449, 520, 521, 523, 524, 525, 526, 527], "b1": 179, "b2": 179, "b_": [376, 382], "b_stride": 1, "ba": [494, 496], "back": [6, 117, 122, 239, 335, 524], "backend": [1, 9, 133, 134, 522], "backward": [1, 517, 520], "bad": 523, "balanc": 464, "baltimor": 214, "bandwidth": [517, 518], "base": [0, 2, 4, 158, 224, 226, 255, 421, 434, 488, 490, 496, 512, 514, 517, 521], "base_idx": 1, "basi": 512, "basic": [5, 282, 520], "batch": [6, 15, 96, 158, 159, 179, 180, 233, 266, 361, 363, 364, 365, 366, 367, 368, 370, 371, 376, 382, 412, 417, 435, 523], "batch_idx": 1, "batch_iter": [7, 490], "batch_siz": [7, 490], "batchnorm": 356, "becaus": [6, 181, 356, 517, 518, 519, 523], "becom": 133, "been": [0, 2, 6, 182, 523], "befor": [1, 2, 6, 9, 29, 154, 156, 254, 335, 396, 434, 501, 518, 521, 523], "before_1": 253, "before_2": 253, "before_i": 253, "before_n": 253, "beforehand": 252, "beggin": 281, "begin": [87, 183, 205, 258, 376, 382, 432, 450, 461, 468, 475, 481, 482, 518], "behav": 122, "behavior": [167, 212, 266, 464, 521, 523], "behaviour": [122, 204, 205], "behind": 520, "being": [308, 356], "bell": 2, "below": [2, 9, 214, 258, 323, 325, 347, 435, 518, 523], "bench": 2, "benchmark": [2, 517], "benefici": [370, 371, 523], "benefit": 518, "best": 518, "beta": [0, 2, 15, 126, 258, 361, 377, 381, 383, 468, 490, 494, 495, 496, 497], "beta_": 2, "beta_1": [492, 494, 495, 496, 497], "beta_2": [494, 495, 496, 497], "better": [335, 499, 520, 527], "between": [0, 2, 8, 101, 175, 434, 457, 460, 461, 464, 508, 518, 522, 523, 524, 527], "beyond": [281, 506, 509], "bfloat16": [2, 12, 187, 347, 524], "bfloat16_t": 2, "bia": [6, 126, 155, 180, 258, 259, 340, 356, 363, 364, 365, 366, 367, 368, 376, 382, 383, 385, 397, 399, 409, 412, 415, 417, 494, 495, 496, 501, 520], "bias": [0, 126, 180, 258, 259, 376, 382, 397, 409, 412], "bias_correct": [494, 495], "bicub": 435, "big": [1, 335, 517], "bigger": [6, 492], "bilinear": [1, 435], "binari": [222, 284, 285, 286, 287, 288, 329, 432, 456, 482, 517, 522], "binary_cross_entropi": [356, 517], "bind": 522, "bit": [0, 126, 180, 201, 258, 259, 280, 336, 347, 392, 414, 415, 416], "bitwis": [0, 92, 93, 94, 95, 201, 280], "bitwise_and": 0, "bitwise_invert": 0, "bitwise_or": 0, "bitwise_xor": 0, "bla": 9, "block": [0, 2, 6, 96, 434], "block_masked_mm": 0, "block_siz": [0, 96], "bn": 361, "bodi": [1, 154, 156], "bool": [0, 1, 2, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 58, 59, 60, 61, 62, 66, 79, 80, 82, 84, 86, 102, 106, 110, 117, 118, 119, 120, 121, 133, 134, 150, 152, 154, 156, 158, 179, 180, 192, 198, 204, 205, 214, 218, 219, 220, 222, 228, 232, 234, 236, 237, 239, 242, 256, 259, 307, 311, 328, 336, 361, 363, 364, 365, 366, 367, 368, 376, 377, 381, 382, 383, 385, 392, 396, 397, 399, 404, 406, 409, 410, 411, 412, 415, 417, 421, 426, 434, 435, 456, 459, 492, 494, 495, 498, 499, 505], "bool_": [12, 347], "boolean": [0, 17, 86, 159, 192, 193, 194, 195, 196, 197, 198, 229, 230, 231, 347, 408, 521], "both": [1, 2, 14, 92, 94, 95, 138, 139, 143, 185, 186, 198, 201, 202, 203, 214, 227, 235, 243, 245, 249, 255, 262, 276, 280, 310, 336, 358, 359, 360, 381, 382, 388, 389, 390, 490, 517, 518, 519, 520, 525, 527], "bottom": 435, "bound": [0, 269, 272, 273, 374, 443, 517, 521, 527], "boundari": 508, "bracket": 6, "brain": 347, "break": 524, "bregler": 370, "bridg": 518, "broadcast": [0, 2, 14, 17, 92, 94, 95, 97, 98, 101, 138, 139, 143, 159, 178, 185, 186, 192, 201, 202, 203, 227, 233, 235, 243, 245, 249, 255, 257, 261, 262, 266, 269, 272, 273, 276, 280, 310, 315, 332, 412], "broadcast_arrai": [0, 2], "broadcast_to": 0, "broadcasted_input": 2, "brought": 8, "btl_tcp_if_includ": [518, 522], "btl_tcp_link": [518, 522], "buffer": [1, 2, 181, 524], "bui": 6, "build": [1, 3, 4, 6, 8, 439, 488, 517, 519], "build_ext": [2, 9], "build_shared_lib": [2, 9], "built": [1, 2, 4, 9, 523], "bundl": 6, "byte": [53, 64, 181, 182, 183, 289, 292, 293, 335, 347, 522], "c": [0, 1, 2, 6, 15, 214, 361, 363, 364, 365, 366, 367, 368, 370, 371, 381, 382, 524, 525, 527], "c_": [382, 497], "c_in": [107, 108, 109, 110, 111, 112, 113], "c_out": [107, 108, 109, 110, 111, 112, 113], "c_pad": 1, "c_t": [382, 497], "cabl": 518, "cach": [6, 9, 100, 181, 182, 289, 517], "calcul": [214, 215, 456, 459, 465, 492], "call": [2, 3, 6, 7, 32, 100, 133, 136, 177, 183, 190, 335, 356, 373, 397, 409, 414, 423, 488, 490, 501, 517, 518, 519, 520, 522, 523], "callabl": [102, 122, 150, 152, 154, 156, 184, 190, 199, 327, 330, 331, 336, 337, 339, 340, 341, 342, 392, 393, 396, 404, 417, 423, 434, 436, 437, 438, 439, 440, 441, 442, 443, 491, 492, 493, 494, 495, 496, 497, 498, 499, 504, 505, 506, 507, 508, 509, 510], "can": [1, 2, 3, 4, 6, 8, 9, 14, 19, 69, 83, 87, 88, 92, 94, 95, 102, 128, 129, 130, 138, 139, 143, 146, 150, 151, 158, 159, 179, 185, 186, 190, 201, 202, 203, 214, 227, 235, 243, 245, 249, 255, 258, 261, 262, 269, 272, 273, 276, 280, 285, 293, 310, 321, 326, 327, 335, 342, 356, 359, 360, 373, 374, 389, 390, 396, 409, 414, 423, 435, 458, 485, 488, 490, 500, 501, 514, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528], "cannot": [6, 101, 521, 524], "captur": [2, 3, 102, 122, 240, 241, 356, 517], "care": [6, 518, 519, 522, 523], "carefulli": [517, 519], "carri": 2, "cartesian": 237, "case": [2, 6, 131, 132, 133, 135, 136, 137, 162, 166, 168, 170, 171, 172, 173, 175, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 215, 216, 217, 218, 219, 220, 233, 279, 305, 326, 359, 360, 370, 389, 390, 432, 450, 468, 475, 481, 482, 500, 501, 517, 518, 519, 520, 522, 525, 526, 527, 528], "cast": [2, 38, 171, 172, 173, 222, 335, 392, 404, 524], "caster": 2, "categor": 6, "categori": [12, 198, 347], "caus": [356, 517, 523], "causal": [6, 159], "caution": 87, "cd": [3, 9], "cdf": [263, 374, 446], "cdot": [448, 457, 460, 477], "ceil": 0, "ceildiv": 1, "cell": 382, "celu": 356, "center": 163, "certain": [406, 517], "chang": [87, 88, 102, 150, 152, 190, 291, 329, 410, 415, 435, 461, 468, 517, 524], "channel": [1, 107, 108, 109, 110, 111, 112, 113, 361, 363, 364, 365, 366, 367, 368, 370, 371], "channel_idx": 1, "char": 2, "charact": 339, "check": [0, 2, 9, 86, 117, 134, 198, 208, 210, 239, 399, 410, 411, 518, 519, 520, 521], "checklist": [518, 522], "checkout": [3, 517], "checkpoint": [434, 490], "chen": 497, "child": 411, "children": 356, "chip": 9, "choleski": 205, "choos": [6, 158, 421, 522], "chosen": 141, "clamp": 175, "clang": 9, "clarifi": 518, "clariti": 520, "class": [2, 6, 7, 10, 11, 12, 31, 122, 130, 174, 344, 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, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 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, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 504, 505, 512], "class_pred": 336, "classif": [439, 440], "classifi": 7, "classmethod": [414, 415], "clear": 100, "click": 9, "clip": [0, 338, 456, 492], "clip_threshold": 492, "clipped_grad": 338, "clone": 9, "close": [5, 8, 9, 17, 192], "closer": 340, "cmake": [3, 4, 9], "cmake_arg": [3, 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, 122, 426, 520], "code": [1, 154, 156, 517, 518, 519, 523], "coeffici": [2, 491, 492, 494, 495, 496, 497], "col": [106, 323], "cold": 9, "collect": [340, 341, 516], "column": [106, 153, 188, 207, 208, 258], "com": 9, "combin": [6, 219, 342], "come": [2, 6, 518, 520], "command": [2, 3, 4, 9, 518, 522], "command_buff": 2, "common": [490, 517, 523], "commonli": [7, 410, 485, 517], "commun": [8, 130, 133, 134, 335, 522], "communication_stream": 335, "communication_typ": 335, "compact": 213, "compar": [2, 86, 517], "comparison": [17, 143, 185, 186, 202, 203, 249], "compat": [6, 150, 152, 159, 190, 262, 266, 374, 526], "compil": [0, 1, 3, 4, 8, 9, 129, 142, 154, 156, 518, 519, 520, 523], "compiled_fun": [517, 519], "compiled_grad_fn": 517, "complement": 93, "complet": [5, 6, 9, 410, 411, 519, 520, 527], "complex": [2, 51, 67, 104, 105, 169, 170, 171, 172, 173, 189, 207, 208, 209, 210, 267, 274, 339, 347, 356, 411, 517, 519, 520], "complex64": [2, 12, 207, 209, 347], "complex64_t": 2, "complexflo": 12, "compon": [2, 4, 6, 163, 219], "compos": [8, 356, 517, 520, 525], "composit": 525, "compress": 288, "compromis": 6, "comput": [0, 1, 2, 5, 6, 7, 8, 9, 118, 119, 120, 121, 122, 126, 141, 149, 158, 184, 199, 200, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 217, 218, 220, 227, 228, 236, 252, 258, 276, 300, 307, 308, 318, 327, 328, 330, 337, 356, 361, 376, 377, 381, 382, 383, 397, 410, 415, 416, 421, 434, 437, 438, 439, 440, 447, 448, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 490, 491, 492, 494, 495, 496, 497, 503, 517, 518, 519, 520, 525, 527], "computation": 523, "compute_encod": 2, "compute_uv": 219, "concaten": [0, 6, 131, 335], "concept": 488, "concis": 6, "concret": [2, 376, 382, 385, 417, 523, 527], "conda": [9, 518], "condit": [0, 332, 517, 527], "config": [2, 4, 518], "configu": 490, "configur": 518, "confirm": [518, 522], "confus": 7, "conj": 105, "conjug": [0, 104], "connect": [499, 518, 522], "consecut": [158, 258, 421], "consequ": 6, "consid": [6, 17, 86, 106, 192, 339, 340, 341, 377, 498, 516, 518], "consider": 517, "const": [0, 1, 2, 459], "constant": [0, 2, 6, 9, 122, 155, 157, 253, 356, 361, 377, 383, 416, 459, 469, 504, 506, 517, 519, 524], "constant_valu": 253, "constitut": 340, "construct": [0, 2, 7, 46, 127, 178, 212, 250, 319, 333], "consult": 518, "consum": 523, "contain": [2, 6, 9, 29, 30, 72, 102, 128, 141, 168, 169, 170, 179, 180, 207, 208, 214, 229, 230, 231, 258, 302, 332, 335, 338, 356, 396, 398, 399, 405, 434, 465, 485, 488, 517, 520], "content": [9, 396, 517], "context": [309, 519], "contigu": [0, 1, 2, 87, 154, 156], "continu": [362, 444, 518, 520], "contract": [0, 141], "contribut": 2, "contriv": [520, 527], "control": [0, 384, 514, 523], "conv": 114, "conv1d": [0, 356], "conv2d": [0, 356], "conv3d": [0, 356], "conv_gener": 0, "conv_transpose1d": 0, "conv_transpose2d": 0, "conv_transpose3d": 0, "conveni": [1, 2, 7, 198], "convent": [19, 114, 140, 141, 435], "convers": 8, "convert": [0, 1, 2, 82, 89, 90, 91, 125, 175, 260, 414, 415, 523, 524, 525], "convolut": [0, 107, 108, 109, 110, 111, 112, 113, 114, 363, 364, 365, 366, 367, 368, 370, 371, 499], "convolv": [107, 108, 109, 110, 111, 112, 113], "convtranspose1d": 356, "convtranspose2d": 356, "convtranspose3d": 356, "cooperative_group": 154, "coordin": [0, 237], "copi": [0, 1, 2, 6, 8, 106, 254, 301, 521, 524], "core": [1, 2, 3, 4, 5, 6, 7, 336, 356, 358, 359, 360, 361, 381, 388, 389, 390, 399, 402, 404, 407, 414, 415, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 465, 485, 488, 490, 517, 518, 524, 525], "corner": 435, "correct": [2, 9, 494, 495, 496, 521, 523], "correctli": [39, 518], "correl": [110, 370], "correspond": [0, 1, 2, 16, 18, 82, 101, 126, 128, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 207, 208, 232, 234, 242, 256, 298, 299, 311, 318, 326, 331, 340, 498, 518, 520, 522], "cos_first": 426, "cosh": [0, 464], "cosin": [0, 20, 21, 115, 116, 457, 506, 508, 520], "cosine_decai": [490, 508], "cosine_similarity_loss": 356, "cost": [9, 492, 518, 523], "costli": 523, "cot": 1, "cot_index": 1, "cotan": 122, "cotang": [1, 2, 122, 330], "could": [6, 356, 521], "count": [356, 508], "counter": 514, "cours": 520, "coursera": 504, "cout": [4, 519], "cov": 266, "covari": [266, 361], "cover": 2, "cpp": [2, 4], "cpu": [8, 207, 208, 209, 210, 216, 347, 527], "cpython": 2, "crash": [87, 517], "creat": [0, 1, 2, 6, 9, 87, 133, 153, 188, 309, 356, 488, 490, 508, 517, 518, 519, 521, 522, 524], "create_additive_causal_mask": 6, "cross": [7, 110, 456, 458], "cross_entropi": [7, 356], "crowd": 6, "cry": 6, "cubic": 435, "cuda": [8, 154], "cummax": 0, "cummin": 0, "cumprod": 0, "cumsum": 0, "cumul": [0, 87, 118, 119, 120, 121, 228], "current": [6, 8, 9, 87, 96, 109, 112, 113, 126, 137, 182, 238, 258, 342, 356, 492, 518, 523], "current_binary_dir": 2, "custom": [8, 122, 154, 156, 434], "custom_decod": 434, "custom_encod": 434, "custom_funct": 1, "custom_kernel_myexp_float": 1, "custom_tim": 2, "cvpr": 370, "cxx": 4, "cycl": 516, "d": [0, 1, 2, 6, 109, 113, 127, 128, 158, 159, 191, 214, 233, 237, 252, 314, 321, 323, 324, 325, 343, 365, 368, 371, 376, 382, 417, 491, 494, 496, 527], "d1": 527, "d2": 527, "d2fdx2": 520, "d_i": 385, "dampen": 505, "darwin": 2, "data": [0, 2, 7, 8, 11, 19, 135, 153, 171, 172, 178, 188, 221, 246, 250, 263, 272, 321, 323, 329, 333, 371, 436, 437, 438, 439, 440, 441, 442, 443, 517, 518, 519, 521, 524], "dataset": [5, 518, 523], "datatyp": 53, "dbuild_shared_lib": 9, "dcmake_build_typ": [4, 9], "ddof": [0, 79, 84, 307, 328], "deal": 517, "deb": 9, "debug": [1, 3, 518, 522], "debugg": 8, "decai": [492, 495, 497, 499, 505, 506, 507, 510], "decay_r": [492, 507, 510], "decay_step": 506, "decent": 7, "decid": [340, 396], "decim": [0, 70, 282], "declar": 2, "decltyp": 1, "decod": 434, "decomposit": [204, 205, 215, 219], "decor": [1, 122], "decoupl": 495, "dedic": 518, "deep": [361, 437, 438, 439, 440], "def": [1, 2, 5, 6, 7, 122, 150, 152, 154, 156, 327, 356, 488, 517, 518, 519, 520, 521, 523, 524, 527], "default": [1, 2, 9, 15, 16, 17, 18, 19, 27, 28, 29, 30, 86, 87, 96, 102, 103, 107, 108, 109, 110, 111, 112, 113, 122, 123, 124, 126, 127, 128, 131, 132, 133, 135, 136, 137, 150, 152, 153, 154, 156, 158, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 175, 179, 180, 184, 187, 188, 192, 200, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 232, 234, 236, 237, 242, 246, 250, 253, 254, 256, 258, 259, 261, 262, 263, 265, 266, 267, 268, 269, 271, 272, 273, 277, 279, 282, 289, 290, 291, 292, 293, 301, 302, 305, 306, 307, 309, 311, 313, 318, 320, 321, 322, 323, 324, 325, 326, 327, 328, 331, 333, 335, 336, 339, 347, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 372, 375, 376, 378, 381, 382, 384, 385, 388, 389, 390, 392, 397, 399, 404, 406, 409, 410, 411, 412, 413, 414, 415, 417, 421, 426, 430, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 488, 491, 492, 493, 494, 495, 496, 497, 499, 504, 505, 506, 514, 516, 517, 518, 519, 520, 522, 524, 526, 528], "default_devic": 528, "default_stream": 528, "defin": [1, 2, 5, 6, 7, 9, 122, 136, 154, 156, 180, 206, 214, 259, 336, 339, 522, 524], "definit": [122, 204, 205, 266], "degre": [0, 260, 469], "deleg": 498, "delta": [461, 491], "delv": [439, 440], "demonstr": 524, "denomin": [381, 457, 491, 493, 494, 495, 496, 504], "dens": [237, 527], "depend": [0, 2, 3, 4, 5, 9, 82, 214, 376, 382, 417, 517, 518, 521, 526, 527], "depth": [339, 360, 365, 368, 371, 390, 520], "dequant": [0, 258], "deriv": [2, 519, 520, 523], "descend": 394, "descent": [505, 517, 523], "describ": [2, 258, 523], "descript": [2, 4, 6, 347], "design": [1, 5, 8, 514, 527], "destin": [0, 2, 63, 137, 244, 257, 339, 490, 519], "destroi": 517, "detach": 520, "detail": [1, 2, 11, 258, 289, 356, 370, 421, 426, 435, 437, 438, 439, 440, 491, 493, 494, 496, 497, 518, 521, 525], "detect": 517, "determin": [0, 2, 128, 266, 342, 347, 403, 526], "dev": [2, 9], "develop": [2, 4, 9], "developer_dir": 9, "deviat": [0, 267, 307, 437, 439, 442], "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, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 123, 124, 125, 126, 127, 128, 131, 132, 135, 136, 137, 138, 139, 140, 143, 144, 145, 147, 148, 149, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 177, 178, 179, 180, 185, 186, 187, 188, 189, 191, 192, 193, 194, 195, 196, 197, 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, 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, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 282, 283, 290, 291, 292, 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, 328, 329, 332, 333, 334, 344, 527, 528], "device_info": 293, "devicetyp": 10, "df": 524, "dfdx": [519, 520, 521], "dft": [160, 161, 162, 164, 165, 166, 171, 172, 173], "dhwc": 371, "diag": [0, 219], "diagon": [0, 46, 127, 153, 321, 323, 324, 325], "dict": [88, 102, 146, 151, 222, 238, 285, 286, 287, 336, 338, 339, 343, 402, 407, 410, 411, 488, 500, 501, 503, 516, 520, 526], "dict_kei": [340, 501], "dictionari": [6, 102, 150, 190, 222, 238, 285, 286, 338, 339, 342, 356, 396, 405, 410, 411, 502, 516, 526], "did": 6, "diff": 2, "differ": [8, 167, 198, 207, 209, 310, 329, 468, 498, 499, 517, 518, 519, 520, 522], "differenti": [1, 8, 362, 444], "difficult": 520, "difficulti": [437, 438], "dilat": [0, 107, 108, 109, 110, 111, 112, 113, 363, 364, 365, 366, 367, 368], "dim": [1, 6, 158, 159, 373, 377, 381, 383, 412, 414, 416, 421, 426, 434], "dimens": [0, 1, 2, 6, 16, 18, 27, 28, 65, 72, 82, 89, 90, 91, 102, 108, 109, 110, 112, 113, 128, 148, 158, 159, 169, 170, 172, 173, 175, 179, 180, 191, 204, 205, 207, 208, 209, 210, 211, 212, 214, 215, 216, 219, 220, 232, 233, 234, 236, 242, 256, 257, 258, 262, 271, 307, 311, 315, 318, 322, 328, 361, 363, 364, 365, 366, 367, 368, 370, 371, 375, 376, 377, 381, 382, 383, 412, 416, 417, 421, 434, 435, 449, 458, 499, 517, 520], "dimension": [31, 155, 157, 160, 161, 162, 164, 165, 166, 171, 172, 173, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 373, 385, 388, 389, 390, 414, 415, 426, 521, 524], "dir": 4, "direct": [6, 394, 497, 527], "directli": [2, 6, 87], "directori": [2, 4, 6, 9], "disabl": [129, 289, 335, 517, 518], "disable_compil": 517, "disappoint": 6, "discard": [6, 339], "discov": [9, 518], "discoveri": 497, "discret": [114, 160, 161, 162, 164, 165, 166, 171, 172, 173, 373, 414], "discuss": 2, "disk": 6, "dispatch": 2, "dispatch_thread": 2, "dispatchthread": 1, "displai": 356, "distanc": [6, 469], "distribut": [8, 9, 261, 262, 263, 265, 266, 267, 272, 273, 335, 385, 437, 438, 439, 440, 442, 443, 459, 462, 467, 469, 485], "distributed_config": [518, 522], "diverg": 462, "divid": [0, 2, 39, 177, 258, 276, 518], "divis": [0, 138, 177, 258, 276], "divisor": [307, 328], "divmod": 0, "dloss_dw": 520, "dloss_dx": 520, "dlpack": 524, "dlvalu": 327, "dmlx_build_cpu": 9, "dmlx_build_cuda": 9, "dmlx_build_gguf": 9, "dmlx_build_safetensor": 9, "dmlx_metal_debug": 3, "dmlx_metal_jit": 9, "do": [0, 2, 6, 9, 212, 329, 356, 398, 409, 485, 488, 517, 518, 519, 520, 523], "doc": [2, 7, 518, 522], "document": [2, 3, 4, 69, 83, 156, 285, 286, 347, 517, 518, 519, 520, 521], "doe": [0, 2, 3, 6, 9, 181, 258, 329, 338, 356, 517, 518, 521, 522, 523, 524], "doesn": [2, 356, 519], "domain": 272, "don": [1, 9, 106, 517, 527], "done": [356, 369, 416, 517, 518, 523, 524], "dot": [151, 211, 220, 318, 339, 401, 412, 518], "doubl": [0, 6, 347], "doubt": 6, "down": [6, 338], "download": 9, "downsampl": [358, 359, 360, 388, 389, 390], "dparam": 327, "dpkg": 9, "draw": 262, "driver": 9, "drop": 396, "dropout": [356, 370, 371, 406, 434, 517], "dropout2d": 356, "dropout3d": 356, "dst": 137, "dt": 144, "dtype": [0, 1, 2, 6, 12, 19, 31, 38, 39, 82, 85, 135, 136, 153, 154, 156, 174, 175, 178, 188, 198, 200, 207, 208, 209, 210, 214, 216, 221, 250, 263, 265, 266, 267, 269, 272, 273, 298, 299, 321, 323, 326, 329, 333, 335, 347, 404, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 465, 506, 507, 508, 509, 510, 517, 518, 519, 520, 521, 524, 525, 526], "dtypecategori": [198, 347], "dual": 464, "duchi": 493, "duplic": 519, "dure": [3, 102, 292, 369, 370, 371, 435, 524], "dx": 122, "dy": 122, "dyld": 518, "dyld_library_path": 518, "dylib": 2, "dynam": [0, 154, 519, 523], "e": [2, 7, 9, 122, 144, 156, 179, 180, 199, 294, 361, 363, 364, 365, 366, 367, 368, 370, 371, 377, 381, 383, 397, 416, 454, 455, 478, 479, 484, 490, 493, 499, 517, 519, 523, 528], "e2m1": 258, "e5": 347, "e8": 347, "each": [0, 1, 2, 72, 88, 126, 146, 158, 180, 198, 204, 205, 207, 208, 209, 210, 211, 215, 219, 220, 233, 237, 253, 258, 259, 262, 277, 287, 288, 302, 319, 322, 329, 331, 332, 370, 371, 373, 376, 377, 382, 417, 421, 434, 456, 458, 514, 517, 518, 519, 522, 523], "eager": 523, "earli": 370, "eas": 6, "easi": [2, 356, 498], "easier": [1, 151, 523], "easiest": 518, "edg": [101, 253, 435, 517], "edit": [9, 411], "effect": [370, 517, 523], "effici": [6, 8, 179, 370, 421, 518, 523, 525], "eigenvalu": [207, 208, 209, 210], "eigenvector": [207, 208], "einstein": [140, 141], "einsum": 141, "either": [9, 14, 69, 82, 83, 92, 94, 95, 101, 138, 139, 143, 177, 185, 186, 190, 201, 202, 203, 214, 227, 233, 235, 243, 245, 249, 255, 258, 276, 280, 310, 327, 359, 360, 389, 390, 423, 435, 439, 440, 518, 522, 524], "elem": [1, 154, 156], "elem_to_loc": [1, 2], "element": [0, 1, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 29, 74, 87, 92, 93, 94, 95, 99, 115, 116, 118, 119, 120, 121, 126, 138, 139, 143, 144, 145, 147, 149, 153, 176, 177, 180, 185, 186, 192, 193, 194, 195, 196, 197, 201, 202, 203, 223, 224, 225, 226, 227, 228, 229, 230, 231, 235, 237, 243, 245, 247, 249, 254, 255, 258, 259, 275, 276, 277, 280, 281, 283, 294, 295, 296, 297, 303, 304, 310, 314, 316, 317, 320, 327, 329, 332, 362, 369, 370, 371, 376, 380, 382, 391, 413, 417, 421, 425, 444, 451, 452, 454, 455, 470, 471, 473, 474, 477, 478, 479, 480, 517, 520, 521], "elementwis": [1, 104, 105], "elif": 6, "ellipsi": 521, "elman": 417, "els": [0, 2, 6, 356, 397, 518, 523], "elsewher": [323, 521], "elu": [356, 475], "emb": [6, 373, 414, 426], "embed": [6, 336, 356, 414, 421, 426, 457, 499], "empti": [266, 339], "en0": 522, "en2": 518, "enabl": [3, 6, 9, 102, 142, 335, 499, 505], "enclos": 519, "encod": [2, 158, 421, 426, 434, 458], "encount": [2, 520], "end": [117, 128, 205, 239, 258, 281, 335, 376, 382, 432, 450, 461, 468, 475, 481, 482, 506, 509, 519], "end_axi": [0, 50, 175], "end_encod": 2, "endif": 2, "endl": [4, 519], "endswith": 397, "enhanc": [6, 421, 523], "enough": [2, 523], "ensur": [0, 1, 2, 9, 154, 156, 338, 464, 518, 519], "ensure_row_contigu": [1, 154, 156], "enter": 6, "entir": [16, 18, 27, 28, 232, 234, 236, 242, 256, 307, 311, 328, 370, 371], "entri": [0, 268, 326, 370, 371], "entropi": [7, 456, 458], "enumer": 356, "environ": [9, 129, 142, 518], "ep": [5, 155, 157, 361, 377, 381, 383, 416, 457, 459, 469, 490, 491, 492, 493, 494, 495, 496, 504], "epoch": 7, "epsilon": [361, 377, 381, 383, 416, 457, 459, 491, 493, 494, 495, 496, 504], "epsilon_1": 492, "epsilon_2": 492, "equal": [0, 1, 17, 29, 86, 153, 186, 192, 203, 249, 254, 269, 302, 335, 381, 385], "equal_nan": [0, 17, 86, 192], "equat": [140, 141, 217, 218], "equival": [0, 2, 32, 69, 83, 136, 139, 154, 177, 180, 187, 314, 362, 372, 374, 378, 379, 380, 386, 387, 411, 413, 415, 418, 419, 420, 422, 424, 427, 428, 429, 430, 431, 433, 518], "erf": [0, 145, 517], "erfinv": 0, "error": [0, 2, 9, 133, 144, 145, 293, 302, 374, 446, 447, 448, 464, 466, 517, 520, 522, 524], "error_norm": 5, "estim": 496, "eta": 497, "etc": [2, 258, 356, 435, 518], "ethernet": [518, 522], "eval": [2, 3, 5, 6, 7, 356, 488, 490, 517, 518, 519, 520, 523, 525], "eval_cpu": 2, "eval_fn": 7, "eval_gpu": 2, "evalu": [2, 6, 7, 8, 88, 137, 146, 199, 292, 330, 356, 395, 406, 488, 490, 517, 519, 525], "even": [1, 2, 6, 102, 167, 207, 209, 517, 518, 519, 523, 524], "evenli": [0, 221], "everi": [1, 258, 340, 490, 510, 520, 522], "everyth": [6, 518], "everywher": 0, "exact": [447, 448], "exactli": [2, 6, 158, 399, 520], "exampl": [0, 3, 4, 5, 6, 7, 9, 19, 39, 88, 122, 133, 150, 151, 152, 154, 156, 158, 159, 175, 190, 198, 200, 207, 208, 209, 210, 214, 216, 298, 299, 309, 314, 326, 338, 341, 342, 356, 358, 359, 360, 361, 381, 388, 389, 390, 397, 399, 406, 409, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 465, 485, 490, 501, 506, 507, 508, 509, 510, 514, 520, 521, 522, 523, 524, 525, 526], "exce": [335, 338], "exceed": 292, "except": [8, 122, 153, 168, 169, 171, 172, 173, 292, 347, 377, 399, 519, 521, 524], "exclud": [257, 315], "exclus": [0, 87, 95], "execut": [2, 4, 9, 89, 90, 91, 183, 200, 518, 524, 527], "execute_process": 4, "exist": [2, 3, 6, 397, 409, 518], "exp": [0, 1, 88, 149, 154, 156, 227, 232, 263, 300, 362, 372, 425, 444, 445, 462, 475, 476, 480, 517, 519, 527], "exp_elementwis": [1, 154, 156], "expand_dim": 0, "expect": [6, 158, 363, 364, 365, 366, 367, 368, 369, 370, 371, 426, 434, 459, 517, 518, 521], "expens": 434, "expensive_fun": 523, "experiment": [88, 150, 152, 190, 524], "explain": 2, "explicit": [2, 501, 514, 524], "explicitli": [179, 356, 514, 522], "explor": 9, "expm1": 0, "exponenti": [0, 147, 149, 362, 372, 422, 444, 445, 475, 507], "exponential_decai": 490, "export": [8, 9, 150, 151, 190], "export_funct": 519, "ext_modul": 2, "extend": [2, 253], "extens": [8, 222, 240, 403, 526], "extern": 524, "extra": [1, 340, 341, 519], "extract": [0, 6, 46, 127, 128, 298, 356, 396, 488], "extras_requir": 2, "extrem": [521, 523], "ey": [0, 6, 211, 220], "f": [0, 2, 5, 7, 122, 214, 356, 382, 495, 517, 524], "f_jvp": 122, "f_t": 382, "f_vjp": 122, "f_vmap": 122, "face": 6, "factor": [2, 15, 187, 204, 205, 212, 213, 216, 435, 458, 507, 510], "fail": [517, 518, 522], "fall": [2, 122], "fallback": 498, "fals": [0, 1, 2, 6, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 58, 59, 60, 61, 62, 66, 79, 80, 84, 86, 102, 106, 110, 118, 119, 120, 121, 133, 150, 152, 156, 179, 180, 192, 198, 204, 205, 214, 218, 219, 220, 222, 228, 232, 234, 236, 237, 242, 256, 307, 311, 328, 332, 336, 339, 340, 341, 342, 347, 377, 381, 383, 385, 397, 399, 409, 412, 415, 421, 426, 434, 435, 456, 459, 492, 494, 495, 505, 519, 524], "famili": 6, "fan": [437, 438, 439, 440], "fan_in": [437, 438, 439, 440], "fan_out": [437, 438, 439, 440], "far": 490, "fast": [1, 8, 374, 448, 518, 527], "faster": [1, 2, 9, 139, 179, 180, 446, 456, 517, 518, 520], "featur": [1, 8, 107, 108, 109, 110, 111, 112, 113, 158, 361, 376, 377, 381, 382, 383, 385, 415, 416, 417, 421, 434, 435, 517, 518, 523], "feed": 6, "feed_forward": 6, "feedforward": [437, 438], "feel": 6, "fetch": 1, "few": [1, 2, 6, 7, 8, 9, 519, 523, 525], "fewer": 518, "ffn": 6, "ffn_norm": 6, "fft": 8, "fftshift": 167, "fi": 518, "figur": 518, "file": [4, 6, 9, 150, 151, 152, 190, 222, 284, 285, 286, 287, 288, 399, 403, 518, 519, 520, 526], "file_or_weight": 399, "fill": [0, 178, 251, 323, 334, 436, 437, 438, 439, 440, 442, 443], "filter": [0, 114, 363, 364, 365, 366, 367, 368, 392, 396, 498, 499], "filter_and_map": 356, "filter_fn": [392, 396], "final": [2, 4, 5, 6, 7, 187, 499, 506, 509, 518, 522], "find": [2, 4, 5, 9, 518], "find_packag": [2, 4], "finder": 9, "fine": [514, 519, 523], "finetun": 356, "finit": [0, 193, 246], "first": [0, 1, 2, 3, 4, 5, 6, 7, 9, 128, 131, 133, 175, 184, 198, 200, 201, 219, 229, 231, 233, 254, 271, 280, 312, 318, 321, 327, 339, 341, 342, 356, 359, 360, 377, 389, 390, 435, 457, 465, 492, 496, 501, 517, 518, 519, 520, 521, 522, 524, 527], "first_lay": 523, "firt": 517, "fit": [258, 527], "five": 517, "fix": [2, 6, 9, 517, 523], "flag": [4, 9, 179, 517, 524], "flat": [179, 180, 339, 343], "flat_param": 287, "flatten": [0, 29, 30, 118, 119, 120, 121, 214, 228, 252, 254, 257, 277, 281, 301, 314, 315, 320, 339, 499, 517], "flexibl": 8, "flexibli": 411, "flip": [0, 110, 114], "float": [0, 1, 2, 12, 15, 17, 19, 82, 155, 156, 157, 158, 159, 174, 177, 178, 187, 192, 198, 214, 246, 258, 259, 261, 265, 335, 338, 347, 361, 369, 370, 371, 377, 381, 383, 392, 404, 416, 421, 426, 432, 434, 435, 436, 437, 438, 439, 440, 442, 443, 457, 458, 459, 461, 465, 468, 469, 481, 482, 491, 492, 493, 494, 495, 496, 497, 499, 504, 505, 506, 507, 509, 510], "float16": [1, 2, 12, 154, 156, 187, 222, 347, 392, 523, 524], "float16_t": [1, 2], "float32": [0, 1, 2, 12, 19, 153, 154, 156, 159, 187, 188, 198, 208, 210, 214, 216, 221, 250, 263, 265, 266, 267, 272, 273, 299, 323, 333, 347, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 465, 506, 507, 508, 509, 510, 517, 518, 519, 520, 521, 523, 524, 525, 526], "float64": [12, 198, 347, 524], "floor": [0, 1, 177], "floor_divid": 0, "flow": [0, 308, 523], "flush": 2, "fn": [190, 337, 340, 341, 342, 525], "follow": [1, 2, 4, 6, 7, 8, 9, 19, 114, 126, 159, 179, 214, 253, 258, 341, 356, 447, 448, 462, 491, 492, 493, 496, 497, 499, 505, 514, 517, 518, 519, 520, 522, 527], "food": 6, "forc": [6, 7, 106, 356, 518, 525], "forg": 518, "format": [6, 151, 222, 258, 284, 285, 286, 287, 288, 518, 524], "formul": [362, 372], "formula": 468, "forth": [435, 518], "forward": [1, 2, 327, 517, 522, 523], "found": [4, 258, 396], "four": 361, "fourier": [160, 161, 162, 164, 165, 166, 171, 172, 173], "fourth": 519, "frac": [144, 258, 267, 294, 361, 369, 370, 371, 377, 381, 383, 385, 416, 425, 437, 438, 439, 440, 457, 459, 461, 464, 476, 478, 479, 491, 493, 494, 495, 496, 504], "fraction": 19, "framework": 8, "free": 289, "freez": [356, 409, 488], "freq": 158, "frequenc": [158, 163, 421, 426], "frequent": [517, 523], "friend": 6, "fro": 214, "frobeniu": 214, "from": [0, 1, 2, 4, 6, 7, 8, 87, 125, 126, 128, 131, 132, 135, 136, 137, 154, 156, 169, 170, 172, 173, 178, 179, 183, 187, 190, 207, 209, 214, 222, 233, 237, 251, 258, 260, 261, 262, 263, 264, 265, 269, 272, 287, 289, 298, 305, 308, 310, 314, 315, 320, 321, 332, 334, 339, 340, 341, 342, 343, 356, 385, 397, 399, 412, 437, 438, 439, 440, 442, 443, 459, 468, 485, 490, 516, 517, 518, 519, 520, 523, 524, 525, 526, 527], "from_embed": 414, "from_linear": 415, "front": [2, 519], "frozen": [356, 397, 407, 409, 415, 488], "fuction": 139, "full": [0, 1, 2, 7, 69, 83, 114, 156, 212, 300, 410, 411, 459, 498, 517, 518, 519, 523], "full_turn": 426, "fulli": [2, 8, 499, 522, 524, 527], "fun": [102, 150, 152, 184, 199, 327, 330, 331, 517, 519, 521, 523, 527], "fun1": 523, "func": 417, "function": [0, 1, 2, 3, 5, 6, 7, 8, 17, 19, 87, 102, 122, 139, 144, 145, 150, 152, 154, 156, 184, 190, 192, 199, 204, 205, 207, 208, 209, 210, 211, 214, 215, 216, 219, 220, 233, 293, 294, 327, 330, 331, 337, 338, 340, 341, 342, 356, 362, 372, 374, 375, 378, 379, 380, 386, 387, 391, 393, 397, 404, 409, 413, 417, 418, 419, 420, 422, 423, 424, 425, 427, 428, 429, 430, 431, 432, 433, 434, 446, 447, 448, 449, 450, 451, 452, 454, 455, 456, 470, 473, 476, 478, 479, 480, 481, 482, 483, 485, 490, 501, 514, 516, 518, 521, 523, 524, 526], "functionexport": 152, "functool": 517, "further": [2, 9, 520], "fuse": [1, 517], "fusibl": 517, "futur": [6, 88, 150, 152, 190, 415, 521, 523], "fx": 122, "g": [3, 9, 122, 156, 214, 258, 382, 484, 499, 504, 505, 519, 523, 528], "g_t": [382, 491, 493, 494, 495, 496, 497, 504, 505], "gain": [437, 438, 439, 440], "gamma": [361, 377, 381, 383, 416, 437, 438, 439, 440], "gap": 1, "gate": [375, 376, 449], "gather": [0, 131, 179, 180], "gather_mm": [0, 180], "gather_qmm": 0, "gaurante": 329, "gaussian": [5, 374, 446, 447, 448, 459], "gaussian_nll_loss": 356, "gc_func": 434, "gelu": [356, 447, 448, 517], "gelu_approx": [356, 374, 446], "gelu_fast_approx": [356, 374, 446], "geluapprox": 374, "gelufast": 374, "gener": [0, 1, 3, 5, 12, 19, 110, 153, 154, 156, 169, 170, 215, 221, 237, 261, 266, 267, 268, 269, 272, 273, 434, 514, 517, 521, 523, 528], "generate_stub": 9, "geq": [432, 482], "get": [2, 5, 7, 9, 108, 109, 110, 112, 113, 123, 124, 174, 181, 182, 183, 238, 264, 356, 517, 519, 520, 523, 527], "get_cache_memori": 100, "get_command_encod": 2, "get_kernel": 2, "get_librari": 2, "gguf": [9, 222, 285, 526], "gh": 1, "gii": 1, "git": 9, "github": [5, 7, 9, 517], "give": [2, 6, 7, 29, 517], "given": [0, 2, 9, 16, 18, 29, 39, 87, 88, 98, 101, 103, 118, 119, 120, 121, 126, 128, 141, 146, 148, 158, 159, 160, 161, 162, 164, 165, 166, 171, 172, 173, 178, 179, 212, 214, 228, 232, 234, 236, 242, 246, 248, 256, 266, 268, 269, 281, 282, 289, 291, 300, 302, 307, 311, 313, 319, 320, 321, 323, 324, 325, 328, 344, 369, 396, 412, 457, 459, 465, 498], "gix": 1, "gix_mult": 1, "giy_mult": 1, "glibc": 9, "global": [129, 131, 132, 133, 135, 136, 137, 142, 270, 335, 338, 514, 517], "glorot": [437, 438], "glorot_norm": 356, "glorot_uniform": 356, "glu": [6, 356], "gm": 1, "gn": 1, "go": [2, 6, 518, 520], "golub": 214, "good": [2, 9, 490, 517, 518, 522, 527], "goroshin": 370, "gower": 6, "gpu": [1, 3, 8, 9, 238, 347, 521, 527], "gputrac": [3, 240], "grad": [2, 5, 7, 122, 327, 338, 490, 500, 517, 518, 519, 520, 521, 523, 525], "grad_fn": [5, 517, 520], "gradient": [0, 5, 7, 122, 184, 308, 327, 335, 337, 338, 356, 397, 410, 415, 434, 464, 488, 490, 491, 492, 494, 495, 496, 497, 500, 503, 505, 517, 518, 520, 521, 523, 524, 525], "grain": 514, "graph": [2, 6, 7, 8, 151, 292, 519, 520], "great": 3, "greater": [0, 6, 29, 149, 186, 254, 338, 432, 482], "greater_equ": 0, "grep": 9, "grid": [2, 154, 156, 237], "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, 458, 468], "group": [0, 1, 107, 108, 109, 110, 111, 112, 113, 126, 131, 132, 133, 135, 136, 137, 159, 180, 258, 259, 329, 335, 336, 363, 364, 377, 414, 415, 518], "group_dim": 2, "group_siz": [0, 126, 180, 258, 259, 336, 414, 415], "groupnorm": 356, "grow": 523, "gru": 356, "guid": [2, 4, 8, 518, 519], "guidelin": 292, "gw": 1, "h": [1, 2, 4, 108, 109, 112, 113, 214, 361, 364, 365, 367, 368, 370, 371, 376, 382, 417, 520, 523], "h_": [376, 382, 417], "h_in": 1, "h_stride": 1, "h_t": [376, 382, 417], "ha": [2, 3, 6, 7, 8, 9, 82, 102, 128, 137, 168, 169, 171, 172, 173, 182, 184, 204, 205, 207, 208, 209, 210, 211, 215, 219, 220, 237, 262, 361, 376, 382, 385, 417, 488, 490, 517, 518, 519, 521, 522, 523, 525, 527], "had": 6, "hadamard": [0, 187], "hadamard_transform": 0, "half": [19, 269, 273, 421, 523], "halv": [375, 449], "hand": [6, 520, 523], "handi": 520, "handl": [2, 356, 517], "happen": [2, 6, 155, 157, 434, 490, 517, 523], "happi": 6, "hard": 6, "hard_shrink": [356, 378], "hard_tanh": [356, 379], "hardcod": 517, "hardshrink": [356, 450], "hardswish": 356, "hardtanh": [356, 451], "hat": [126, 258], "have": [0, 1, 2, 6, 9, 17, 86, 89, 90, 91, 96, 122, 131, 159, 169, 170, 172, 173, 180, 192, 233, 240, 258, 262, 329, 335, 339, 382, 412, 423, 497, 501, 516, 517, 518, 519, 521, 522, 523, 527], "haven": 6, "hazan": 493, "he": [6, 439, 440], "he_norm": 356, "he_uniform": 356, "head": [159, 412, 434], "header": [2, 9, 154, 156], "heart": 6, "heavi": 6, "height": [359, 360, 361, 364, 365, 367, 368, 370, 371, 389, 390], "hello": [339, 343], "help": [2, 6, 517, 527], "helper": [6, 154, 156, 335, 517, 518, 522], "henc": [0, 2, 258, 517], "hendryck": 448, "here": [2, 6, 9, 490, 517, 519, 520, 523, 526, 527], "hermitian": [208, 210], "hf": 382, "hg": 382, "hh": 417, "hi": [6, 382], "hidden": [376, 382, 417, 434, 499], "hidden_dim": [7, 488, 490], "hidden_s": [376, 382, 417], "hierarchi": 347, "high": [269, 273, 356, 373, 443, 485], "high_pad_s": 0, "higher": [2, 191, 293, 465, 518, 520], "highli": 9, "him": 6, "hing": 460, "hinge_loss": 356, "hinton": 504, "hit": 2, "hn": 376, "ho": 382, "hold": [2, 6, 11, 12, 214, 517], "homebrew": 518, "hopkin": 214, "host": 2, "host1": 518, "host2": 518, "host3": 518, "host4": 518, "host_nam": 1, "hostfil": [518, 522], "hostnam": [518, 522], "hostname1": [518, 522], "hostname2": [518, 522], "hostname3": 518, "hostname4": 518, "hot": 458, "hour": 6, "how": [2, 4, 6, 7, 356, 358, 359, 360, 363, 364, 365, 366, 367, 368, 373, 388, 389, 390, 414, 435, 500, 517, 521, 527], "howev": [2, 122, 356, 374, 377, 501, 514, 517, 518, 523, 524], "hr": 376, "http": [9, 377, 381, 383, 391, 416, 448, 470], "huber": 461, "huber_loss": 356, "human": [439, 440], "hundr": 9, "hurri": 6, "hutter": 495, "hyperbol": [0, 21, 23, 26, 116, 297, 317, 433, 483], "hz": 376, "i": [0, 1, 2, 3, 4, 6, 7, 8, 9, 17, 19, 29, 38, 82, 87, 88, 101, 108, 109, 110, 112, 113, 114, 117, 118, 119, 120, 121, 122, 127, 128, 131, 132, 134, 135, 136, 137, 139, 146, 150, 152, 154, 155, 156, 157, 158, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 175, 177, 178, 179, 180, 187, 190, 192, 193, 198, 199, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 222, 227, 228, 232, 233, 237, 239, 253, 254, 257, 258, 259, 266, 267, 268, 279, 281, 284, 285, 286, 292, 293, 294, 300, 302, 307, 308, 313, 314, 315, 318, 321, 322, 326, 327, 328, 329, 330, 331, 332, 335, 336, 338, 339, 340, 341, 342, 347, 349, 356, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 369, 370, 371, 374, 376, 377, 381, 382, 383, 385, 388, 389, 390, 396, 397, 403, 405, 406, 408, 409, 410, 411, 412, 413, 415, 416, 417, 421, 426, 432, 434, 435, 439, 440, 446, 448, 456, 457, 459, 464, 465, 468, 469, 471, 477, 482, 488, 490, 492, 494, 495, 497, 498, 500, 501, 506, 508, 509, 514, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528], "i386": 9, "i_n": 1, "i_nw": 1, "i_s": 1, "i_sw": 1, "i_t": 382, "iclr": [494, 495, 496], "id": [7, 9], "idea": [520, 523], "idempot": [397, 409], "ident": [0, 122, 137, 153, 167, 308, 356, 406, 518], "identifi": [2, 339, 516], "idim": 7, "idiom": [7, 517], "idx": [39, 521], "ie": 409, "ieee": 347, "ifac": 518, "ignor": [6, 39, 88, 101, 102, 146, 492, 522], "ih": 417, "ii": 1, "ij": 237, "im": 267, "imag": [0, 364, 365, 367, 368, 370, 371, 435], "imagenet": [439, 440], "imaginari": [51, 189], "immedi": [6, 392], "implement": [0, 1, 5, 7, 158, 159, 179, 180, 373, 396, 412, 421, 423, 426, 432, 434, 435, 482, 491, 492, 493, 496, 497, 499, 500, 512, 517, 520], "impli": 329, "implicit": [514, 517, 520], "implicitli": 523, "import": [2, 3, 5, 6, 7, 9, 122, 133, 187, 190, 214, 287, 327, 339, 340, 341, 342, 343, 356, 358, 359, 360, 361, 381, 388, 389, 390, 399, 435, 456, 458, 465, 485, 488, 490, 517, 518, 520, 521, 523, 524, 525], "import_funct": 519, "imported_ab": 519, "imported_fun": 519, "imported_funct": 519, "improv": [1, 2, 3, 6, 456, 491, 492, 493, 494, 495, 496, 504, 517, 518], "in_ax": [331, 520], "in_channel": [363, 364, 365, 366, 367, 368], "in_dim": [356, 488], "in_proj": 488, "includ": [1, 2, 4, 118, 119, 120, 121, 151, 154, 156, 181, 182, 228, 292, 383, 393, 405, 415, 459, 490, 517, 519, 520, 521, 525, 526, 528], "include_dir": 2, "inclus": [0, 42, 43, 44, 45, 58, 118, 119, 120, 121, 175, 228], "incom": 2, "inconveni": 517, "incorpor": 524, "incorrect": 524, "increas": [293, 522], "increment": 19, "incur": [6, 9], "independ": [130, 370, 371], "index": [0, 1, 2, 8, 10, 29, 39, 148, 153, 179, 184, 237, 254, 298, 299, 314, 315, 327], "indic": [0, 2, 17, 27, 28, 29, 30, 39, 159, 179, 180, 184, 192, 193, 194, 195, 196, 197, 198, 212, 219, 257, 298, 299, 302, 314, 315, 327, 406, 408, 458, 465, 508, 521], "indices_or_sect": [75, 302], "indirectli": 524, "individu": [356, 370, 371], "ineffici": [521, 523], "inexact": [12, 198], "inf": [214, 246, 412], "infer": [8, 178, 222, 321, 326, 518, 519], "infin": [0, 194, 196, 197, 246, 388, 389, 390, 496], "infinit": [17, 192, 193], "info": [6, 9], "inform": [3, 4, 6, 7, 9, 141, 174, 238, 285, 286, 347, 356, 361, 374, 412, 518, 519, 520, 527], "inherit": [7, 516], "inifn": 194, "init": [356, 413, 485, 490, 506, 507, 509, 510, 518], "init_fn": [436, 437, 438, 439, 440, 441, 442, 443, 485], "init_valu": 1, "initi": [1, 3, 5, 6, 133, 342, 356, 361, 377, 381, 383, 385, 413, 416, 436, 437, 438, 439, 440, 441, 442, 443, 488, 501, 506, 507, 509, 510, 517, 518, 519, 523], "initializer_list": 0, "inject": 0, "inlin": 0, "inner": [0, 517], "inorm": 381, "inp": [1, 154, 156], "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, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 122, 125, 127, 128, 131, 132, 137, 138, 139, 140, 141, 143, 144, 145, 147, 148, 149, 150, 151, 152, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 177, 179, 180, 184, 185, 186, 187, 189, 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, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 242, 243, 244, 245, 246, 247, 249, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 268, 271, 274, 275, 276, 277, 279, 280, 281, 282, 283, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 314, 315, 316, 317, 318, 319, 320, 321, 322, 324, 325, 326, 327, 328, 329, 331, 332, 334, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 370, 371, 373, 375, 376, 377, 381, 382, 383, 385, 388, 389, 390, 412, 415, 416, 417, 421, 432, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 459, 460, 461, 462, 464, 465, 467, 469, 482, 485, 517, 519, 520, 521, 522, 525, 526], "input_dil": [0, 110], "input_dim": [7, 356, 385, 415], "input_nam": [1, 154, 156], "input_s": [376, 382, 417], "inputs1": 465, "inputs2": 465, "insert": [128, 148, 527], "insid": [517, 519], "inspect": [3, 517, 525], "inspir": 8, "instabl": 469, "instal": [2, 4, 522], "instanc": [6, 39, 122, 258, 343, 356, 381, 392, 393, 394, 397, 399, 400, 401, 406, 409, 410, 411, 423, 488, 518, 522, 524], "instancenorm": 356, "instanti": [1, 2, 7, 523], "instantiate_kernel": 2, "instead": [2, 9, 122, 356, 411, 426, 520, 523], "instruct": [4, 9, 519], "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, 58, 59, 60, 61, 62, 63, 66, 70, 72, 75, 78, 79, 80, 81, 82, 84, 87, 96, 98, 103, 107, 108, 109, 110, 111, 112, 113, 118, 119, 120, 121, 126, 127, 128, 135, 136, 137, 141, 148, 153, 154, 158, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 178, 180, 181, 182, 183, 184, 188, 198, 206, 214, 221, 228, 232, 234, 236, 238, 242, 244, 250, 253, 254, 256, 257, 258, 259, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 277, 279, 281, 282, 289, 292, 293, 298, 299, 300, 301, 302, 305, 306, 307, 311, 312, 314, 315, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 331, 333, 335, 336, 356, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 373, 375, 376, 377, 381, 382, 383, 385, 388, 389, 390, 412, 414, 415, 416, 417, 421, 426, 434, 449, 457, 458, 462, 467, 469, 488, 499, 506, 508, 509, 510], "int16": 347, "int32": [0, 1, 12, 19, 39, 175, 198, 200, 214, 269, 298, 326, 347, 435, 521, 525], "int64": [12, 347], "int64_t": 2, "int8": [12, 347], "int_0": 144, "integ": [0, 12, 177, 179, 180, 198, 214, 238, 253, 258, 259, 261, 268, 269, 302, 314, 318, 331, 347, 373, 404, 508, 521], "integr": [19, 314, 523], "intend": [0, 517], "interact": 434, "interest": 527, "interfac": [2, 518, 522], "intermedi": 524, "intern": 361, "interpol": 435, "interpret": 4, "interv": [19, 221, 269, 273], "introduc": [0, 281], "intuit": 356, "invalid": [0, 87], "invers": [0, 20, 21, 22, 23, 24, 25, 26, 93, 145, 164, 165, 166, 167, 168, 169, 170, 205, 211, 215, 220], "invert": 0, "involv": [490, 517], "iogpu": 293, "iostream": 4, "ip": [518, 522], "ip1": [518, 522], "ip2": [518, 522], "ip3": 518, "ip4": 518, "is_avail": 133, "is_equival": 2, "is_leaf": [339, 340, 341, 342], "is_leaf_fn": 396, "isclos": 0, "isfinit": 0, "ish": 6, "ishmael": 6, "isinf": 0, "isnan": 0, "isneginf": 0, "isposinf": 0, "issu": [518, 520, 524], "issubdtyp": [2, 12, 347], "item": [0, 2, 5, 6, 7, 340, 519, 523, 524, 525], "iter": [5, 7, 219, 340, 341, 499, 514, 517, 523], "iterm": 9, "itertool": [6, 340], "its": [0, 1, 2, 9, 159, 205, 215, 233, 254, 271, 323, 337, 343, 356, 415, 490, 494, 495, 496, 518, 523, 524, 527], "itself": [2, 336, 498, 501], "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, 214, 370, 493, 494, 496], "j8": 2, "jacobian": [2, 199, 330, 525], "jain": 370, "jax": [8, 514], "jit": [1, 154, 156], "jmlr": 493, "jnp": 524, "john": 214, "join": 508, "join_schedul": 490, "jointli": 266, "json": [518, 522], "just": [2, 4, 7, 383, 517, 519, 521], "jvp": [2, 122, 525], "k": [0, 6, 46, 96, 107, 111, 127, 153, 159, 179, 187, 320, 323, 324, 325, 385, 397], "kaim": 440, "kd": [109, 113], "keep": [2, 16, 18, 27, 28, 232, 234, 236, 242, 256, 307, 311, 328, 356, 396, 520, 523], "keepdim": [0, 16, 18, 27, 28, 34, 35, 36, 37, 59, 60, 61, 62, 66, 79, 80, 84, 214, 232, 234, 236, 242, 256, 300, 307, 311, 328], "kei": [1, 3, 6, 150, 159, 190, 238, 261, 262, 263, 265, 266, 267, 268, 269, 271, 272, 273, 339, 340, 396, 397, 409, 412, 501, 514, 516, 519, 520], "kept": 293, "kernel": [2, 8, 9, 107, 108, 109, 110, 111, 112, 113, 154, 156, 358, 388, 517, 521], "kernel_dil": [0, 110], "kernel_s": [358, 359, 360, 363, 364, 365, 366, 367, 368, 388, 389, 390], "key_cach": 6, "key_input_dim": 412, "key_proj": 6, "keyring_1": 9, "keyword": [150, 184, 287, 288, 327, 340, 356, 514, 519, 526, 528], "kh": [108, 109, 112, 113], "kind": 6, "kingma": [494, 496], "kl_div_loss": 356, "kname": 2, "know": [2, 6], "known": [424, 477], "kron": 0, "kroneck": [0, 200], "kth": [0, 29, 254], "kullback": 462, "kw": [108, 109, 112, 113], "kw_onli": 2, "kwarg": [11, 130, 150, 151, 190, 287, 288, 344, 519, 528], "l": [6, 7, 107, 111, 204, 205, 208, 210, 212, 356, 361, 363, 366, 376, 382, 417, 468], "l1": [327, 461, 463, 464, 468], "l1_loss": 356, "l2": [461, 464, 499, 505], "l2_loss": 356, "l_": 461, "la": 214, "label": [3, 5, 458, 465], "label_smooth": 458, "lack": 521, "lambd": [378, 430, 450, 481], "lambda": [340, 341, 342, 356, 378, 392, 397, 404, 430, 450, 475, 481, 491, 492, 493, 494, 495, 496, 497, 504, 505, 517, 518, 519, 520], "languag": [1, 2, 4], "lapack": 9, "larg": [6, 356, 412, 464, 517, 519, 523], "larger": [1, 158, 293, 421, 497], "largest": [214, 246, 320], "lasso": 327, "last": [0, 1, 6, 30, 82, 155, 157, 162, 166, 169, 170, 172, 173, 175, 179, 180, 183, 191, 204, 205, 207, 208, 209, 210, 211, 215, 216, 219, 220, 233, 262, 301, 318, 329, 363, 364, 365, 366, 367, 368, 370, 371, 377, 435, 498, 499, 524], "later": [3, 9, 490], "launch": [1, 2, 133, 518, 521], "layer": [8, 155, 336, 356, 358, 359, 360, 370, 371, 376, 377, 382, 383, 385, 388, 389, 390, 406, 411, 414, 415, 417, 423, 434, 484, 488, 499, 519, 522], "layer_s": 7, "layernorm": 356, "layout": 1, "lazi": [8, 488, 525], "lazili": [6, 356], "lceil": 96, "ld": [376, 382, 417], "lead": [0, 19, 87, 517], "leaf": [102, 336, 339, 340, 341, 342, 396], "leaf_modul": 356, "leaki": [384, 453], "leaky_relu": 356, "leakyrelu": 356, "learn": [5, 7, 8, 361, 377, 381, 383, 413, 416, 490, 491, 492, 493, 494, 495, 496, 497, 499, 504, 505], "learnabl": [363, 364, 365, 366, 367, 368, 423], "learning_r": [7, 490, 491, 492, 493, 494, 495, 496, 497, 499, 501, 504, 505, 506, 507, 508, 509, 510, 517], "least": [6, 89, 90, 91, 101, 158, 204, 205, 207, 208, 209, 210, 211, 215, 216, 219, 220, 258], "leav": [2, 88, 146, 340, 341, 342], "lectur": 504, "lecun": 370, "left": [0, 6, 158, 201, 214, 258, 281, 374, 421, 435, 447, 448, 459, 461, 469], "left_shift": 0, "leibler": 462, "len": [6, 162, 166, 170, 173, 187, 508], "length": [6, 158, 167, 305, 361, 363, 366, 376, 382, 417, 508], "leq": [461, 475], "less": [0, 1, 6, 29, 203, 254, 293, 335, 421, 468, 498, 518], "less_equ": 0, "let": [1, 2, 5, 6, 205, 517, 519, 520, 523, 524], "level": [0, 179, 180, 439, 440], "lh": [376, 382, 417], "lhs_indic": [0, 179, 180], "lhs_mask": 96, "lib": [2, 518], "libbla": 9, "libcudnn9": 9, "liblapack": 9, "libmlx": 9, "libmlx_ext": 2, "libmpi": 518, "librari": [1, 2, 4, 9, 349, 356, 518, 519], "like": [2, 6, 8, 136, 150, 152, 190, 198, 251, 334, 371, 464, 501, 503, 517, 518, 519, 520, 522, 523, 524, 525, 527], "likelihood": [459, 467], "limit": [0, 2, 101, 289, 292, 293, 521], "linalg": 187, "line": [6, 518, 519, 522, 523, 524], "linear": [0, 2, 6, 7, 8, 217, 218, 336, 340, 356, 362, 372, 374, 375, 384, 399, 415, 417, 418, 420, 422, 424, 435, 444, 445, 446, 447, 448, 449, 453, 472, 474, 475, 477, 485, 488, 501, 508, 509, 517, 519], "linear1": 6, "linear2": 6, "linear3": 6, "linear_schedul": [490, 508], "linearli": 412, "link": [2, 4, 9], "linspac": 0, "lion": 490, "list": [1, 6, 11, 16, 18, 31, 75, 82, 87, 88, 89, 90, 91, 98, 102, 103, 110, 141, 146, 154, 156, 161, 162, 163, 165, 166, 167, 169, 170, 172, 173, 178, 184, 199, 214, 232, 234, 236, 237, 242, 250, 253, 256, 261, 262, 263, 265, 266, 267, 269, 272, 273, 285, 300, 302, 306, 307, 311, 318, 319, 322, 327, 328, 330, 333, 339, 342, 343, 356, 397, 399, 400, 401, 402, 407, 409, 410, 411, 488, 494, 495, 496, 497, 498, 508, 516, 517, 518, 519, 520, 522, 523], "list_or_scalar": 82, "listen": 518, "liter": [2, 253, 435, 439, 440, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469], "littl": 6, "liu": 6, "live": [8, 154, 156, 527], "ll": [1, 5, 7, 461, 517, 520], "llama": 6, "llamaattent": 6, "llamaencoderlay": 6, "llm": 8, "load": [2, 7, 8, 349, 399, 518], "load_weight": [356, 523], "loader": 7, "loader_path": 2, "loan": 214, "loc": [1, 265, 267], "local": [356, 370, 518], "localhost": [518, 522], "locat": [0, 2, 4, 87, 298, 299, 410, 411, 518, 521, 527], "log": [0, 225, 227, 232, 386, 387, 454, 455, 456, 459, 462, 464, 467, 480], "log10": 0, "log1p": 0, "log2": 0, "log_cosh_loss": 356, "log_sigmoid": [356, 386], "log_softmax": [356, 387], "logaddexp": 0, "logarithm": [0, 223, 224, 225, 226], "logcosh": 464, "logcumsumexp": 0, "logic": [0, 2, 229, 230, 231, 518], "logical_and": 0, "logical_not": 0, "logical_or": 0, "logist": [0, 5, 294, 448, 477], "logit": [6, 262, 456, 458, 517], "logsigmoid": 356, "logsoftmax": 356, "logsumexp": [0, 228], "long": 6, "longer": [6, 114, 520], "look": [2, 6, 518], "lookup": 373, "loop": [6, 7, 517, 518, 520, 523], "loshchilov": 495, "loss": [5, 7, 327, 356, 490, 517, 518, 520, 523], "loss_and_grad": 356, "loss_and_grad_fn": [7, 490, 517, 520], "loss_fn": [5, 7, 490, 517, 520], "loss_grad_fn": 518, "lot": 520, "low": [269, 273, 443, 485], "low_pad_s": 0, "lower": [204, 205, 208, 210, 218, 220, 258, 269, 272, 273, 323, 443], "lr": [5, 497], "lr_schedul": [506, 507, 508, 509, 510], "lstm": 356, "lto": 2, "lu": [6, 213], "luckili": 523, "lvalu": 327, "m": [0, 2, 4, 6, 9, 96, 153, 179, 187, 214, 323, 491, 517], "m1": [1, 6, 517, 520, 527], "m10": 347, "m7": 347, "m_": [494, 495, 496, 497], "m_t": [494, 495, 496, 497], "mac": 518, "machin": [6, 8, 9, 504, 518], "maco": [9, 293], "macosx": 9, "made": [6, 349], "mai": [2, 4, 88, 150, 152, 179, 180, 190, 214, 336, 370, 499, 518, 520, 521], "main": [4, 8, 128, 153, 154, 156, 321, 340, 341, 356, 518], "maintain": [370, 371, 497], "major": [0, 2, 106], "make": [1, 2, 3, 4, 6, 7, 9, 106, 151, 152, 233, 248, 291, 356, 498, 506, 507, 509, 510, 517, 523, 525, 527], "make_shar": 2, "malloc": 2, "man": 6, "manag": [8, 309, 514, 518, 519, 527], "mani": [1, 2, 87, 302, 363, 364, 365, 366, 367, 368, 373, 414, 517, 518, 519, 523], "manual": [356, 518], "map": [2, 7, 39, 222, 340, 373, 392, 519], "map_fn": [392, 396], "map_torch_to_mlx": 6, "margin": [465, 469], "margin_ranking_loss": 356, "mask": [0, 6, 96, 159, 406, 412, 521], "mask_lh": [0, 96], "mask_n": 1, "mask_nw": 1, "mask_out": [0, 96], "mask_rh": [0, 96], "mask_s": 1, "mask_sw": 1, "matadata": 222, "match": [9, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 181, 399, 435, 458, 521, 524], "materi": [6, 8], "math": [6, 469, 517], "mathbf": 205, "mathcal": [267, 385], "mathemat": 214, "mathrm": [144, 294, 381], "matmul": [0, 179, 527], "matric": [214, 216, 219], "matrix": [0, 5, 15, 46, 96, 126, 127, 153, 179, 180, 187, 188, 204, 205, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 219, 220, 233, 237, 258, 259, 266, 414, 415, 441, 485], "matter": [6, 356, 519], "matur": 518, "max": [0, 1, 2, 214, 235, 362, 388, 389, 390, 413, 444, 451, 452, 457, 459, 460, 465, 469, 471, 473, 474, 492, 496, 517, 520, 527], "max_buffer_s": 238, "max_freq": 426, "max_i": 258, "max_norm": 338, "max_recommended_working_set_s": [238, 293], "max_val": 451, "maximum": [0, 7, 27, 39, 101, 118, 183, 292, 338, 356, 384, 388, 389, 390, 418, 426, 447, 448, 453, 472, 488, 523], "maxpool1d": 356, "maxpool2d": 356, "maxpool3d": 356, "maxtotalthreadsperthreadgroup": 2, "mca": [518, 522], "md": 214, "me": 6, "mean": [0, 1, 5, 6, 7, 154, 157, 265, 266, 267, 327, 356, 361, 377, 397, 416, 442, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 517, 518, 520, 524], "meant": 122, "measur": 527, "mechan": 434, "medic": 371, "meet": 9, "member": [356, 402, 407], "memori": [0, 1, 2, 8, 87, 100, 154, 181, 182, 183, 278, 289, 292, 293, 434, 488, 492, 517, 523, 524], "memory_order_relax": 1, "memory_s": [238, 293], "memoryview": [523, 524], "merg": 517, "meshgrid": 0, "metadata": [5, 222, 285, 286], "metal": [2, 4, 8, 154, 156, 292], "metal_captur": 3, "metal_kernel": 1, "metal_path": 9, "metallib": [2, 9], "method": [2, 6, 10, 11, 31, 122, 130, 174, 336, 344, 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, 403, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 501, 504, 505, 512], "millisecond": [9, 517, 527], "min": [0, 2, 214, 243, 362, 413, 444, 451, 452, 471, 474], "min_freq": 426, "min_i": 258, "min_val": 451, "mind": 6, "mine": 6, "minibatch": 7, "minim": [518, 522], "minimum": [0, 28, 39, 101, 119, 426, 456, 457], "minsizerel": 9, "minu": 149, "minut": 6, "mish": 356, "mismatch": 519, "miss": [399, 519, 526], "mix": 521, "mkdir": [3, 9], "ml": 9, "mlp": [7, 356, 434, 490], "mlp_dim": [6, 434], "mlx": [1, 3, 5, 6, 7, 9, 349, 356, 485, 488, 490, 514, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "mlx_build_acceler": 4, "mlx_build_benchmark": 9, "mlx_build_cpu": 9, "mlx_build_cuda": 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": [129, 142, 517], "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": [150, 152, 190, 519], "mnist": 7, "mode": [0, 1, 2, 114, 126, 180, 253, 258, 259, 336, 395, 406, 408, 414, 415, 435, 439, 440], "model": [5, 7, 8, 287, 336, 337, 340, 341, 356, 392, 395, 397, 399, 403, 406, 408, 409, 410, 412, 434, 485, 488, 490, 500, 501, 503, 517, 518, 519, 523], "modest": 2, "modif": 524, "modifi": 524, "modul": [2, 4, 6, 7, 336, 337, 423, 434, 485, 503, 516, 517, 523], "moment": [6, 492, 496, 518], "momentum": [361, 497, 499, 501, 505, 517], "monei": 6, "monitor": 522, "monoton": 470, "moor": 215, "more": [1, 2, 3, 4, 7, 11, 82, 128, 150, 179, 204, 205, 207, 208, 209, 210, 211, 212, 215, 219, 220, 233, 258, 285, 286, 289, 292, 347, 356, 361, 370, 421, 426, 434, 435, 437, 438, 439, 440, 456, 514, 517, 518, 520, 521, 525, 527], "moreov": 522, "most": [2, 159, 262, 326, 356, 503, 517, 518, 520, 521, 523], "move": [0, 2, 244, 527], "moveaxi": 0, "mpi": [133, 349], "mpirun": [518, 522], "mse": 327, "mse_loss": 356, "mtl": 2, "mtl_capture_en": 3, "mtlcommandbuff": 2, "mu": 505, "much": [1, 2, 6, 358, 359, 360, 388, 389, 390, 517, 523], "multi": [8, 159, 363, 364, 365, 366, 367, 368, 519, 521, 524], "multidimension": 237, "multiheadattent": [6, 356], "multioptim": 490, "multipl": [0, 1, 9, 15, 96, 152, 155, 157, 179, 180, 233, 245, 259, 412, 426, 507, 508, 510, 517, 523, 526], "multipli": [0, 2, 39, 180, 259, 369, 426, 435], "muon": 490, "murtadha": 6, "must": [0, 1, 2, 3, 9, 96, 101, 150, 158, 159, 178, 180, 208, 210, 214, 258, 261, 262, 266, 269, 272, 273, 332, 435, 524], "mutat": 521, "mx": [1, 2, 3, 4, 5, 6, 7, 39, 88, 104, 105, 122, 133, 136, 150, 151, 152, 154, 156, 159, 175, 190, 198, 200, 207, 208, 209, 210, 212, 214, 216, 222, 268, 287, 298, 299, 326, 327, 338, 356, 358, 359, 360, 361, 372, 381, 384, 388, 389, 390, 392, 399, 403, 418, 435, 436, 437, 438, 439, 440, 441, 442, 443, 445, 453, 456, 457, 458, 462, 465, 472, 483, 485, 488, 490, 514, 517, 518, 519, 520, 521, 523, 524, 525, 526, 527, 528], "mxfp4": [126, 258], "my": [6, 9], "my_devic": 528, "my_path": 287, "my_script": [518, 522], "myexp": [1, 154, 156], "myexp_strid": 1, "mymlp": 488, "n": [0, 1, 2, 6, 31, 96, 107, 108, 109, 110, 111, 112, 113, 153, 159, 160, 162, 164, 166, 168, 171, 173, 187, 188, 266, 267, 307, 323, 328, 361, 363, 364, 365, 366, 367, 368, 370, 371, 376, 382, 417, 435, 464, 469, 518, 522], "n_kv": 159, "n_q": 159, "n_t": 376, "naiv": [2, 520], "naive_add": 520, "name": [1, 2, 122, 151, 154, 156, 180, 222, 259, 285, 286, 287, 288, 356, 377, 396, 399, 401, 518, 521, 526], "named_modul": 356, "namespac": 4, "nan": [0, 17, 86, 192, 193, 195, 246], "nan_to_num": 0, "nanobind": 2, "nanobind_add_modul": 2, "nativ": [9, 518], "natur": [0, 223, 225, 523], "nb": 2, "nb_domain": 2, "nb_modul": 2, "nb_static": 2, "nbyte": 2, "nc": 361, "nccl": 133, "ndarrai": [31, 521, 523, 525], "ndhwc": [365, 368, 371], "ndim": [0, 1, 2, 175, 214, 219, 435], "ne": 1, "nearest": [1, 435], "necessari": [106, 356], "necessarili": [209, 320], "need": [1, 2, 4, 6, 7, 8, 9, 86, 356, 410, 411, 426, 434, 514, 518, 520, 522, 523, 524, 525, 527], "neg": [0, 128, 175, 196, 246, 281, 321, 384, 388, 389, 390, 412, 459, 467, 469, 521], "negat": [0, 247], "negative_slop": [384, 453], "neginf": [0, 246], "neighbor": [435, 522], "neither": [184, 327], "nelem": 2, "nervou": 6, "nest": [82, 102, 342, 356, 488, 516, 520], "nesterov": [499, 505], "network": [6, 8, 335, 361, 370, 373, 437, 438, 485, 488, 499, 504, 518], "neural": [6, 8, 373, 437, 438, 470, 485, 488, 499, 504], "never": [6, 523], "new": [0, 1, 2, 7, 98, 128, 244, 248, 279, 306, 322, 329, 340, 341, 404, 412, 488, 490, 503, 508, 517, 519, 521, 523, 524], "new_tre": 341, "newton": 499, "next": [2, 4, 6, 7, 289, 519], "nh": [376, 382, 417], "nhwc": [361, 364, 367], "nice": [520, 523], "nlc": [361, 363, 366], "nld": [376, 382, 417], "nlh": [376, 382, 417], "nll": [459, 467], "nll_loss": 356, "nn": [2, 6, 7, 287, 340, 356, 485, 488, 490, 501, 503, 517, 519, 523], "nobodi": 6, "node": [88, 102, 146, 331, 341, 342, 518, 522], "nois": 5, "noisi": 5, "nomins": 2, "non": [0, 1, 2, 4, 9, 237, 407, 417, 470, 488], "nondeterminist": 521, "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, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 86, 87, 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, 118, 119, 120, 121, 122, 125, 126, 127, 128, 129, 131, 132, 135, 136, 137, 138, 139, 140, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 153, 155, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 184, 185, 186, 187, 188, 189, 191, 192, 193, 194, 195, 196, 197, 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, 240, 241, 242, 243, 244, 245, 246, 247, 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, 276, 277, 278, 279, 280, 281, 282, 283, 284, 286, 290, 291, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 331, 332, 333, 334, 335, 336, 339, 340, 341, 342, 358, 359, 360, 374, 388, 389, 390, 392, 396, 397, 404, 409, 412, 417, 426, 434, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 492, 512, 519, 521], "nonlinear": [417, 517], "nonzero": 521, "noop": [409, 518], "nor": [2, 184, 327], "norm": [6, 157, 338, 377, 469, 496, 497], "norm1": 6, "norm2": 6, "norm_first": 434, "normal": [1, 2, 5, 6, 154, 155, 156, 157, 159, 207, 208, 266, 272, 356, 358, 359, 360, 361, 377, 381, 383, 388, 389, 390, 416, 434, 437, 439, 524, 527], "not_equ": 0, "notabl": [6, 8], "notat": [126, 339, 401], "note": [0, 1, 2, 4, 6, 9, 17, 19, 87, 96, 102, 109, 112, 113, 122, 126, 169, 170, 180, 181, 192, 212, 214, 258, 262, 329, 336, 356, 416, 435, 490, 521, 524, 526], "noth": [6, 122, 356, 523], "notic": [6, 519, 520, 526], "now": [1, 2, 6, 9, 415, 517, 524], "np": [1, 6, 7, 518, 524, 525], "npy": [222, 284, 526], "npz": [6, 222, 287, 288, 399, 403, 526], "ns_step": 499, "nuc": 214, "nuclear": 214, "nuisanc": 518, "nullopt": 0, "num": [0, 6, 221, 271], "num_class": [7, 490], "num_decoder_lay": 434, "num_embed": [373, 414], "num_encoder_lay": 434, "num_epoch": [7, 490], "num_exampl": 5, "num_featur": [5, 361], "num_group": 377, "num_head": [6, 412, 434], "num_it": 5, "num_lay": [6, 7, 490], "num_param": 356, "num_paramet": 413, "num_sampl": 262, "num_split": 0, "number": [0, 2, 12, 19, 64, 74, 102, 108, 109, 110, 112, 113, 126, 151, 153, 159, 180, 184, 188, 199, 221, 246, 253, 258, 259, 262, 265, 267, 271, 273, 277, 281, 282, 318, 319, 323, 327, 330, 331, 335, 336, 356, 361, 363, 364, 365, 366, 367, 368, 370, 371, 377, 381, 412, 413, 434, 435, 437, 438, 439, 440, 499, 506, 508, 509, 514, 517, 520, 522, 528], "number_of_el": 0, "numer": [6, 155, 157, 214, 227, 232, 300, 361, 377, 381, 383, 416, 456, 457, 459, 469, 491, 492, 493, 494, 495, 496, 504, 517, 523], "numpi": [2, 6, 7, 8, 14, 17, 19, 92, 94, 95, 97, 98, 138, 139, 143, 185, 186, 192, 201, 202, 203, 207, 209, 227, 233, 235, 243, 245, 249, 255, 276, 280, 310, 523, 525, 526], "nvidia": 9, "nw": 1, "nwhc": 370, "o": [9, 159, 382], "o_t": 382, "obj": 285, "object": [3, 11, 31, 102, 151, 152, 154, 156, 198, 331, 339, 340, 341, 342, 347, 370, 434, 516, 522], "observ": 6, "occupi": [126, 180, 258, 259], "occur": 524, "odd": 167, "odim": 7, "odot": [376, 382], "off": [6, 9, 523], "offer": 464, "offset": [0, 1, 2, 6, 47, 87, 128, 155, 158, 321], "often": 371, "ok": [399, 517, 519, 520], "okai": [517, 523], "old": 6, "older": [150, 152, 190], "omit": [496, 518], "onc": [1, 2, 9, 517, 519], "one": [0, 2, 4, 6, 9, 39, 82, 89, 97, 101, 108, 109, 110, 112, 113, 133, 148, 150, 153, 155, 157, 158, 179, 214, 225, 233, 259, 262, 305, 310, 326, 335, 347, 366, 367, 368, 409, 435, 458, 498, 518, 519, 522, 527], "ones": [0, 2, 6, 251, 287, 299, 323, 410, 411, 490, 518, 521], "ones_lik": 0, "onli": [1, 2, 6, 8, 86, 96, 108, 109, 110, 112, 113, 122, 159, 179, 208, 210, 214, 219, 258, 266, 293, 329, 347, 356, 396, 397, 399, 404, 406, 409, 410, 411, 488, 517, 518, 519, 520, 522, 526, 527], "onlin": 493, "op": [1, 2, 252, 329, 397, 523], "open": [3, 9, 19, 269, 273, 518], "openmpi": 518, "oper": [3, 6, 8, 10, 38, 89, 90, 91, 110, 159, 179, 180, 255, 257, 300, 308, 315, 344, 347, 356, 434, 497, 517, 518, 520, 521, 523, 524, 525, 527, 528], "operand": [140, 141, 179], "opportun": 517, "opt": [500, 518], "optim": [1, 3, 5, 7, 8, 410, 517, 518, 520, 523], "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, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 87, 89, 90, 91, 96, 102, 103, 107, 108, 109, 110, 111, 112, 113, 114, 118, 119, 120, 121, 122, 126, 127, 128, 131, 132, 133, 135, 136, 137, 150, 152, 153, 155, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 178, 179, 180, 184, 188, 196, 197, 200, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 228, 232, 234, 236, 237, 242, 246, 250, 253, 254, 256, 258, 259, 261, 262, 263, 265, 266, 267, 268, 269, 271, 272, 273, 277, 279, 281, 286, 300, 301, 302, 305, 306, 307, 311, 313, 314, 318, 320, 321, 322, 323, 324, 325, 326, 327, 328, 331, 333, 335, 336, 339, 340, 341, 342, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 376, 382, 385, 388, 389, 390, 392, 396, 397, 399, 404, 409, 412, 414, 415, 417, 421, 426, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 491, 492, 493, 494, 495, 496, 497, 499, 501, 504, 505, 506, 514, 517, 519, 526, 528], "ord": 214, "order": [0, 1, 29, 87, 110, 141, 208, 209, 210, 214, 254, 320, 356, 377, 410, 423, 501, 517, 520, 522], "ordinari": 191, "org": [377, 381, 383, 391, 416, 448, 470], "origin": [6, 128, 338, 361, 405, 437, 438, 439, 440, 491, 492, 493, 496, 497, 499, 519, 521, 524], "orthogon": 499, "orthonorm": 187, "other": [0, 2, 6, 8, 198, 214, 356, 398, 488, 497, 517, 518, 519, 521, 522, 523, 525], "other_input": 356, "otherwis": [19, 110, 133, 268, 336, 339, 340, 341, 342, 397, 399, 409, 432, 434, 435, 450, 456, 461, 468, 481, 482, 523, 524], "our": [1, 2, 6, 7, 423, 491, 492, 493, 496, 497, 499, 518], "out": [0, 1, 2, 9, 96, 154, 156, 159, 190, 370, 371, 406, 517, 518, 519, 520, 521], "out_ax": [331, 520], "out_channel": [363, 364, 365, 366, 367, 368], "out_dim": [356, 488], "out_dtyp": 2, "out_idx": 2, "out_mask": 96, "out_proj": [6, 488], "out_ptr": 2, "out_shap": [1, 2], "outer": [0, 517, 523], "outlier": 464, "output": [0, 1, 2, 6, 9, 16, 17, 18, 19, 29, 87, 96, 97, 98, 102, 104, 105, 106, 111, 112, 113, 118, 119, 120, 121, 122, 140, 151, 153, 154, 155, 156, 157, 158, 159, 168, 171, 172, 173, 178, 179, 184, 187, 188, 192, 214, 221, 228, 232, 234, 236, 237, 242, 246, 250, 251, 254, 256, 257, 261, 262, 263, 265, 266, 267, 269, 272, 273, 287, 288, 298, 299, 300, 305, 307, 311, 315, 321, 323, 327, 328, 329, 330, 331, 332, 333, 334, 361, 363, 364, 365, 366, 367, 368, 381, 385, 412, 415, 432, 434, 435, 437, 438, 439, 440, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 482, 485, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "output_dim": [7, 356, 385, 415], "output_directori": 2, "output_dtyp": [1, 154, 156], "output_fil": 6, "output_nam": [1, 154, 156], "output_pad": [0, 111, 112, 113, 366, 367, 368], "output_shap": [1, 154, 156], "output_strip_trailing_whitespac": 4, "output_vari": 4, "outsid": [154, 156, 175], "over": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 107, 108, 109, 110, 111, 112, 113, 118, 119, 120, 121, 162, 163, 166, 167, 170, 173, 191, 214, 219, 221, 228, 232, 234, 236, 242, 254, 256, 283, 300, 301, 307, 311, 318, 320, 328, 361, 363, 364, 365, 366, 367, 368, 377, 383, 416, 458, 506, 509, 518, 520, 522], "overal": 2, "overhead": [1, 517, 523, 527], "overlap": 1, "overload": 19, "overrid": [2, 142], "overview": 3, "overwrit": 6, "own": [9, 518, 524], "owndata": 524, "p": [9, 212, 261, 356, 369, 370, 371, 469, 494, 496], "pack": [180, 258, 259], "packag": [2, 5, 7, 9, 349, 485, 518, 522], "package_data": 2, "pad": [0, 1, 107, 108, 109, 110, 111, 112, 113, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 358, 359, 360, 363, 364, 365, 366, 367, 368, 388, 389, 390], "pad_valu": 0, "pad_width": [0, 253], "padding_hi": 0, "padding_lo": 0, "page": [518, 525], "pain": 6, "pair": [0, 2, 253, 399, 421], "pairwis": 469, "pan": 6, "paper": [361, 426, 491, 492, 493, 496, 497], "parallel": [518, 527], "param": [327, 336, 356, 485, 519, 520], "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, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 125, 126, 127, 128, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 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, 175, 176, 177, 178, 179, 180, 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, 240, 242, 243, 244, 245, 246, 247, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 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, 335, 336, 337, 338, 339, 340, 341, 342, 343, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 381, 382, 383, 384, 385, 388, 389, 390, 392, 393, 396, 397, 399, 404, 405, 406, 409, 410, 411, 412, 413, 414, 415, 416, 417, 421, 423, 426, 430, 432, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 482, 484, 485, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 503, 504, 505, 506, 507, 508, 509, 510, 512, 517, 518, 519, 520, 523], "parameter_scal": 492, "parametr": [413, 471], "pars": [6, 151], "parse_arg": 6, "parser": 6, "part": [1, 2, 51, 67, 150, 152, 189, 190, 274, 520, 521], "parti": 518, "partial": [410, 411, 517, 523], "particip": [131, 132, 135, 136, 137], "particular": [258, 377], "particularli": 517, "partit": [0, 29], "pass": [1, 2, 6, 7, 9, 69, 83, 179, 180, 252, 253, 327, 335, 337, 339, 340, 341, 356, 397, 409, 410, 411, 423, 517, 518, 519, 522, 523], "password": [518, 522], "path": [3, 4, 9, 141, 150, 151, 152, 190, 222, 240, 284, 285, 286, 287, 288, 336, 341, 399, 498, 518, 522], "pattern": [356, 523], "peak": [183, 278], "penalti": [499, 505], "penros": 215, "pep": 524, "per": [6, 7, 126, 159, 180, 258, 259, 335, 336, 361, 377, 381, 383, 416, 512, 517, 518, 522, 523], "perceptron": [8, 519], "perf_count": 517, "perfectli": 523, "perform": [0, 1, 2, 3, 6, 8, 15, 96, 110, 118, 119, 120, 121, 137, 140, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 179, 180, 187, 208, 210, 228, 233, 259, 282, 300, 314, 335, 356, 377, 434, 439, 440, 490, 499, 517, 518, 521, 523, 527], "perhap": [2, 6], "perm": 7, "permtuat": 268, "permuat": 212, "permut": [0, 7], "persist": 9, "pg": 214, "phi": [374, 446], "physic": 518, "pi": [144, 374, 426, 447, 520], "pick": 2, "pip": [2, 4, 9], "pipelin": 2, "pivot": [212, 213], "pixel": 370, "place": [6, 39, 281, 282, 336, 518, 523, 524], "placehold": 517, "plai": [2, 6], "plain": 423, "plan": [2, 517], "platform": 9, "plot": 518, "plu": [0, 225], "png": 518, "point": [0, 2, 5, 6, 9, 87, 174, 177, 258, 259, 347], "pool": [358, 359, 360, 388, 389, 390, 527], "popul": 2, "port": 522, "portion": 369, "posinf": [0, 246], "posit": [0, 6, 29, 128, 158, 175, 184, 197, 204, 205, 244, 246, 254, 266, 281, 321, 327, 340, 356, 363, 364, 365, 366, 367, 368, 412, 421, 426, 459, 469, 519], "possibl": [133, 179, 302, 373, 414, 517, 518, 521, 527], "possibli": [1, 6, 15, 96, 179, 233, 338], "postur": 6, "power": [0, 520, 524], "practic": [2, 517], "pre": [9, 159, 456], "preced": 377, "precis": [0, 6, 149, 159, 258, 356, 374, 416, 456, 500, 517], "preclud": 356, "pred": [460, 464], "predic": [336, 404, 498], "predict": [456, 459, 460, 461, 462, 463, 464, 466, 467, 468], "prefix": [331, 339, 341], "prelu": 356, "prepar": [2, 6, 518], "prepend": [3, 233], "preprint": [6, 491, 497], "preprocessor": 9, "present": 1, "preserv": [279, 520], "press": [6, 214], "pretti": [517, 523], "prevent": [308, 469, 524], "previou": [289, 292, 293], "primal": [1, 2, 122, 199, 330], "primit": 520, "print": [1, 2, 5, 6, 7, 9, 88, 122, 200, 338, 339, 340, 341, 343, 356, 514, 517, 518, 519, 520, 521, 522, 523, 524, 525], "prior": [257, 314, 315], "priorit": 520, "privat": [2, 4], "prng": [261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 514], "prob": 456, "probabl": [9, 269, 369, 370, 371, 415, 456, 458, 462, 527], "problem": [5, 7, 356], "process": [6, 110, 114, 130, 131, 132, 133, 135, 136, 137, 335, 340, 341, 371, 373, 434, 516, 518, 522], "processor": 9, "prod": [0, 1], "produc": [0, 2, 9, 102, 412, 485, 519], "product": [0, 2, 15, 87, 120, 191, 199, 200, 206, 233, 252, 256, 318, 330, 412, 525], "profil": 3, "program": [4, 183], "programmat": 411, "project": [3, 4, 6, 412, 519], "project_source_dir": 2, "promot": [2, 159], "promote_typ": 2, "promoted_dtyp": 2, "prompt": 6, "propag": [520, 521], "properti": [32, 39, 48, 51, 53, 64, 65, 67, 72, 74, 405, 408, 502, 518, 520], "proportion": 338, "protocol": 524, "provid": [0, 2, 6, 87, 150, 151, 184, 267, 268, 281, 318, 327, 335, 340, 342, 349, 356, 392, 397, 399, 409, 410, 411, 414, 415, 434, 435, 484, 488, 498, 518, 519, 526, 528], "pseudo": [215, 514], "pth": 6, "public": [2, 356], "pun": 0, "pure": [1, 122, 356, 490], "purpos": [1, 214, 518], "purs": 6, "push": 2, "push_back": 2, "put": [0, 1, 7, 257, 517, 518], "put_along_axi": [0, 212], "py": [2, 6, 9, 518, 522], "pypi": 9, "python": [1, 3, 4, 6, 52, 72, 82, 88, 146, 335, 339, 340, 341, 342, 343, 488, 500, 501, 503, 516, 518, 519, 520, 522, 524], "python_execut": 4, "python_requir": 2, "pytorch": [6, 8, 374, 377, 520], "pytorch_compat": 377, "q": [159, 216], "quantiz": [0, 126, 180, 222, 259, 414, 415], "quantized_matmul": 0, "quantizedembed": 356, "quantizedlinear": 356, "quarter": 6, "queri": [6, 159, 293, 412], "query_input_dim": 412, "query_proj": 6, "question": [6, 523], "queue": 3, "quick": [2, 8], "quit": [520, 524], "quotient": [0, 138, 139, 177], "r": [2, 6, 216, 327, 370, 376], "r_t": 376, "race": 527, "radian": [0, 125], "rag": 6, "rain": 6, "rais": [0, 6, 122, 214, 255, 302, 399, 519], "ram": [6, 292], "random": [1, 2, 3, 5, 6, 7, 8, 154, 156, 159, 358, 359, 360, 361, 381, 388, 389, 390, 399, 406, 517, 519, 520, 527, 528], "randomli": [5, 6, 268, 369, 370, 371], "rang": [0, 2, 3, 5, 6, 7, 9, 19, 175, 179, 221, 438, 440, 447, 448, 490, 506, 507, 508, 509, 510, 514, 517, 520, 523, 527], "rank": [0, 135, 136, 137, 465, 518, 522], "rate": [5, 490, 491, 492, 493, 494, 495, 496, 497, 499, 504, 505], "rather": [2, 520, 527], "ratio": [0, 25], "rceil": 96, "re": [7, 9, 267, 485], "reachabl": 518, "readabl": 3, "real": [0, 168, 169, 170, 171, 172, 173, 204, 205, 207, 208, 209, 210, 267], "realli": 383, "reason": [1, 6, 521], "reboot": 9, "receiv": [135, 136, 336, 508, 518, 524], "reciproc": [0, 283], "reclaim": 289, "recommend": [9, 292, 497, 499], "recompil": [102, 517], "reconstruct": 212, "record": [3, 183, 523], "recreat": [343, 490], "rectifi": [384, 418, 420, 439, 440, 453, 472, 474], "recurr": [376, 382, 417], "recurs": [151, 356, 396, 397, 402, 407, 409, 488], "recv": [136, 518], "reduc": [0, 1, 9, 16, 18, 27, 28, 132, 232, 234, 236, 242, 256, 307, 311, 328, 335, 342, 361, 434, 464], "reduct": [16, 18, 132, 232, 234, 242, 256, 342, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469], "redund": 520, "refer": [214, 381, 391, 405, 437, 438, 439, 440, 448, 470, 521], "reflect": [405, 517, 521, 524], "regard": 374, "regardless": [87, 159, 518], "regist": [2, 7], "regress": [8, 464], "regular": [39, 370, 470, 495, 517, 519, 521], "regularli": 2, "reimplement": 2, "rel": [17, 192, 492, 517, 518], "relative_step": 492, "releas": 4, "relev": 2, "reli": [1, 2], "relu": [356, 413, 434, 471, 485], "relu2": 356, "relu6": 356, "relu\u00b2": [419, 473], "remain": [0, 6, 293, 327, 341, 369, 370, 371, 518], "remaind": [0, 139], "remov": [0, 128, 233, 262, 305, 458], "rep": [0, 319], "repeat": [0, 319], "repeatedli": 5, "repetit": 277, "replac": [0, 6, 246, 410, 411, 434, 468], "replai": 3, "repli": 6, "repo": [5, 7, 9, 517], "report": [181, 292], "repres": [2, 6, 130, 133, 180, 465, 469, 524], "represent": [6, 213, 258, 329, 339, 343], "request": 154, "requir": [1, 2, 4, 6, 356, 518, 522, 523, 524], "requires_grad": 520, "rerun": [517, 523], "rescal": 338, "research": 8, "reset": 278, "reset_peak_memori": 183, "reshap": [0, 6, 214, 435, 517, 521], "resid": 293, "resolv": 2, "resourc": 2, "resource_limit": 238, "respect": [2, 5, 7, 122, 155, 157, 179, 180, 184, 258, 327, 340, 356, 361, 374, 377, 381, 383, 488, 518, 520, 522, 525], "respons": 2, "rest": [6, 158, 340, 341, 421, 522], "restart": 9, "restor": 281, "result": [0, 6, 15, 19, 39, 82, 87, 102, 151, 155, 157, 180, 200, 214, 233, 259, 266, 277, 292, 306, 340, 341, 342, 347, 426, 456, 517, 518, 520, 524], "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, 52, 72, 82, 86, 87, 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, 118, 119, 120, 121, 122, 125, 126, 127, 128, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 143, 144, 145, 147, 148, 149, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 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, 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, 242, 243, 244, 245, 246, 247, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 277, 279, 280, 282, 283, 289, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 336, 337, 338, 339, 340, 341, 342, 343, 356, 376, 382, 392, 393, 394, 396, 397, 398, 399, 400, 401, 402, 406, 407, 409, 410, 411, 417, 436, 437, 438, 439, 440, 441, 442, 443, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 485, 488, 498, 500, 516, 517, 518, 519, 520, 521, 523, 524, 526, 527], "return_metadata": 222, "revers": [0, 2, 42, 43, 44, 45, 58, 87, 118, 119, 120, 121, 228, 322, 426], "rf": 9, "rfft": 168, "rfft2": 169, "rfftn": 170, "rho": 491, "rhs_indic": [0, 179, 180], "rhs_mask": 96, "right": [0, 1, 2, 9, 207, 258, 280, 281, 374, 435, 447, 448, 459, 461, 469], "right_shift": 0, "ring": 133, "rm": [6, 9, 157, 492], "rmsnorm": [6, 356], "rmsprop": 490, "rnn": [356, 376], "robust": 464, "roform": [6, 421], "roll": 0, "root": [0, 6, 157, 283, 303, 416], "rope": [6, 356], "rosetta": 9, "rotari": [6, 158, 421], "rotat": [158, 421], "round": [0, 258], "row": [0, 1, 2, 87, 106, 153, 154, 156, 188, 258, 323], "rpath": 2, "rsqrt": 0, "rtol": [0, 17, 192], "rule": [2, 490], "run": [1, 2, 3, 4, 6, 7, 8, 9, 10, 154, 156, 252, 344, 361, 392, 491, 492, 494, 495, 496, 517, 519, 522, 523, 527, 528], "runtim": [6, 133, 349, 517, 518], "runtime_error": 2, "safetensor": [9, 222, 286, 399, 403, 490, 523, 526], "sai": [2, 6, 485, 523], "said": 6, "sake": 520, "same": [0, 2, 6, 9, 17, 39, 86, 97, 98, 102, 108, 109, 110, 112, 113, 114, 131, 155, 157, 163, 167, 168, 171, 172, 173, 180, 184, 192, 199, 253, 262, 281, 282, 299, 329, 330, 332, 335, 341, 356, 359, 360, 361, 369, 377, 381, 389, 390, 414, 436, 437, 438, 439, 440, 441, 442, 443, 458, 469, 488, 500, 514, 517, 518, 519, 521, 522, 527], "sampl": [2, 5, 6, 221, 261, 262, 263, 265, 266, 269, 272, 273, 437, 438, 439, 440, 442, 443, 459, 465, 469, 514, 517, 519], "sat": 6, "save": [3, 6, 8, 222, 240, 258, 285, 286, 287, 288, 403, 519, 523], "save_gguf": 526, "save_safetensor": [403, 490, 526], "save_weight": 356, "savez": [6, 403, 526], "savez_compress": 526, "saw": [6, 520], "scalar": [0, 2, 14, 15, 17, 31, 52, 82, 86, 92, 93, 94, 95, 96, 98, 101, 138, 139, 143, 158, 177, 178, 184, 185, 186, 187, 192, 201, 202, 203, 221, 227, 229, 230, 231, 233, 235, 243, 245, 246, 249, 253, 255, 261, 267, 269, 272, 273, 276, 280, 285, 310, 327, 329, 332, 337, 469, 519, 520, 523, 525], "scale": [0, 2, 6, 15, 126, 155, 157, 158, 159, 180, 187, 258, 259, 265, 267, 338, 370, 371, 383, 412, 421, 422, 426, 435, 475, 492], "scale_arr": 2, "scale_factor": 435, "scale_paramet": 492, "scatter": 0, "scatter_add": 0, "scatter_add_axi": 0, "scatter_max": 0, "scatter_min": 0, "scatter_prod": 0, "schedul": [2, 490, 506, 507, 508, 509, 510, 512, 527], "schema": [3, 522], "schulz": 499, "scipi": [187, 212], "scope": 356, "score": [6, 159, 465], "script": [518, 522], "sdk": 9, "se": 1, "second": [6, 9, 128, 198, 200, 201, 229, 231, 233, 258, 280, 312, 321, 327, 359, 360, 389, 390, 457, 465, 492, 496, 517, 519, 520, 527], "second_layer_a": 523, "second_layer_b": 523, "secret": 6, "section": [1, 6, 9, 302, 469, 517, 518, 520], "see": [1, 2, 4, 6, 7, 9, 11, 12, 33, 34, 35, 36, 37, 40, 41, 42, 43, 44, 45, 47, 49, 50, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 214, 285, 286, 289, 336, 347, 356, 361, 362, 370, 372, 374, 378, 379, 380, 386, 387, 395, 413, 414, 415, 418, 419, 420, 421, 422, 424, 426, 427, 428, 429, 430, 431, 433, 435, 437, 438, 439, 440, 446, 447, 448, 475, 517, 518, 519, 520, 521, 522, 525, 527], "seed": 264, "seen": [518, 524], "segment": 0, "segmented_mm": 0, "select": [0, 3, 9, 208, 210, 320, 332, 392, 396, 404, 522], "self": [6, 7, 10, 31, 32, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 52, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 122, 174, 356, 470, 488], "selu": 356, "semant": [14, 92, 94, 95, 97, 98, 138, 139, 143, 185, 186, 201, 202, 203, 227, 233, 235, 243, 245, 249, 255, 276, 280, 310, 527], "semi": [204, 205, 266], "send": 518, "sender": 518, "sennrich": 6, "sensit": 464, "sentencepiec": 6, "separ": [0, 6, 69, 83, 377, 465], "sequenc": [6, 16, 18, 34, 35, 59, 60, 61, 62, 66, 78, 79, 80, 84, 87, 98, 110, 135, 148, 154, 156, 158, 161, 162, 163, 165, 166, 167, 169, 170, 172, 173, 178, 184, 232, 234, 236, 242, 250, 256, 261, 262, 263, 265, 266, 267, 269, 272, 273, 279, 298, 299, 300, 302, 305, 307, 311, 318, 319, 322, 326, 327, 328, 333, 361, 363, 366, 376, 382, 417, 434, 514, 527], "sequenti": [356, 485], "seri": 9, "serial": 490, "set": [2, 4, 6, 7, 9, 102, 122, 129, 131, 132, 133, 135, 136, 137, 142, 155, 157, 158, 238, 289, 290, 291, 292, 293, 309, 335, 374, 383, 385, 395, 397, 404, 405, 406, 409, 410, 415, 421, 432, 457, 469, 482, 488, 490, 492, 494, 495, 501, 514, 519, 520, 523], "set_byt": 2, "set_compute_pipeline_st": 2, "set_data": 2, "set_dtyp": 356, "set_input_arrai": 2, "set_memory_limit": 289, "set_output_arrai": 2, "set_vector_byt": 2, "setup": [2, 4, 5, 7, 9, 517, 518, 519], "sever": [6, 9, 107, 108, 109, 110, 111, 112, 113, 287, 288, 335, 517, 518, 522, 526], "sgd": [5, 7, 490, 497, 501, 506, 507, 510, 517], "shade": [1, 2], "shall": 6, "shape": [0, 2, 3, 6, 7, 69, 86, 87, 96, 97, 98, 102, 107, 108, 109, 110, 111, 112, 113, 128, 131, 135, 136, 150, 152, 154, 156, 158, 159, 160, 163, 164, 167, 168, 171, 172, 173, 178, 179, 187, 199, 211, 220, 233, 250, 251, 261, 262, 263, 265, 266, 267, 269, 272, 273, 279, 281, 299, 326, 329, 330, 332, 333, 334, 356, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 370, 371, 376, 381, 382, 385, 388, 389, 390, 399, 417, 436, 437, 438, 439, 440, 441, 442, 443, 458, 469, 490, 517, 519, 520, 521, 525, 527], "shapeless": [0, 102, 150, 152], "share": [8, 126, 154, 180, 258, 259, 329, 518], "shared_memori": 154, "shazeer": 6, "shift": [0, 163, 167, 201, 280, 281, 361], "shop": 6, "should": [1, 2, 4, 5, 6, 7, 9, 87, 100, 128, 131, 154, 155, 156, 157, 159, 199, 240, 257, 258, 293, 315, 321, 327, 330, 335, 336, 339, 356, 363, 364, 365, 366, 367, 368, 370, 371, 406, 412, 423, 458, 460, 465, 488, 498, 499, 516, 517, 518, 519, 520, 523, 524, 528], "show": [9, 347, 517], "shown": 2, "shuffl": 7, "side": [0, 253, 358, 359, 360, 366, 367, 368, 388, 389, 390, 517], "sigma": [374, 375, 376, 382, 425, 437, 438, 439, 440, 448, 449, 454, 476, 477], "sigmoid": [0, 6, 356, 386, 424, 448, 454, 456, 477], "sign": [0, 17, 192, 347, 497], "signal": [114, 435], "signatur": [1, 154, 156], "signedinteg": [12, 198], "signific": 258, "significantli": 518, "silent": [171, 172, 173], "silicon": [2, 6, 8, 9, 527], "silu": 356, "sim": 267, "simd": 1, "simd_sum": 1, "simdgroup": 1, "simdgroup_s": 1, "similar": [6, 180, 198, 340, 410, 411, 412, 457, 518, 524, 526], "similarli": [2, 9, 233, 258, 520, 523], "simpl": [2, 6, 7, 356, 373, 484, 490, 517, 518, 519, 520, 522, 523], "simple_axpbi": 2, "simple_tim": 2, "simplest": [2, 356, 518], "simpli": [2, 6, 9, 372, 384, 418, 445, 453, 472, 483, 488, 517, 518, 520, 522], "simplic": 0, "simplifi": 518, "simultan": 1, "sin": [0, 122, 426, 519, 520, 525], "sinc": [1, 2, 6, 7, 180, 183, 488, 497, 508, 517, 519, 524, 527], "sine": [0, 22, 23, 296, 297, 519, 520], "sing": 214, "singer": 493, "singl": [7, 88, 146, 199, 222, 237, 253, 330, 359, 360, 389, 390, 517, 519, 521, 526], "singleton": [0, 16, 18, 27, 28, 133, 232, 233, 234, 236, 242, 256, 307, 311, 328, 518], "singular": [214, 215, 219], "sinh": 0, "sink": 159, "sinusoid": 426, "sinusoidalpositionalencod": 356, "size": [0, 1, 2, 6, 7, 53, 72, 96, 108, 109, 112, 113, 126, 148, 154, 155, 156, 157, 158, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 178, 180, 182, 187, 188, 198, 206, 214, 258, 259, 262, 279, 292, 293, 298, 302, 305, 326, 329, 335, 336, 356, 358, 359, 360, 363, 364, 365, 366, 367, 368, 373, 381, 388, 389, 390, 414, 415, 435, 492, 518, 523, 524], "size_in_megabyt": 293, "size_t": [0, 2], "skip": [3, 87], "slice": [0, 299, 521], "slice_s": [0, 298], "slice_upd": 0, "slide": [358, 359, 360, 388, 389, 390], "slight": [6, 523], "slightli": [421, 527], "slope": 384, "slow": 517, "slowli": 6, "sm": 9, "small": [6, 149, 155, 157, 335, 361, 377, 383, 416, 459, 464, 469, 517, 527], "smaller": [0, 9, 254, 335, 497, 517], "smallest": 214, "smile": 6, "smooth": [458, 468, 504], "smooth_l1_loss": 356, "sned": 137, "snippet": 518, "so": [1, 2, 6, 9, 184, 187, 327, 369, 435, 490, 517, 518, 521, 523, 527], "socket": 518, "softmax": [0, 6, 159, 356, 387, 455, 458], "softmin": 356, "softplu": [356, 391, 470], "softshrink": 356, "softsign": 356, "solut": [217, 218], "solv": 356, "some": [0, 2, 5, 6, 7, 151, 397, 409, 490, 501, 517, 518, 519, 520, 522, 523], "someon": 6, "someth": [5, 6, 521], "sometim": 517, "sonoma": 9, "soon": 6, "sort": [0, 29, 30, 179, 180, 254, 320], "sorted_indic": [0, 179, 180], "sourc": [0, 1, 2, 3, 4, 63, 135, 136, 154, 156, 244, 322, 518], "space": [0, 2, 221, 456, 467], "spars": [0, 237], "spatial": [108, 109, 110, 112, 113, 358, 359, 360, 377, 388, 389, 390, 435], "speak": [6, 214], "specif": [1, 2, 9, 258, 518, 520], "specifi": [0, 2, 19, 38, 108, 109, 110, 112, 113, 128, 169, 170, 178, 184, 206, 214, 221, 244, 250, 257, 262, 277, 312, 314, 315, 318, 321, 322, 327, 331, 333, 361, 432, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 482, 517, 518, 519, 520, 527], "spectrum": 163, "speed": 1, "spent": 6, "split": [0, 375, 377, 449], "splittabl": 514, "sqrt": [0, 6, 144, 159, 187, 361, 374, 377, 381, 383, 385, 416, 426, 437, 438, 439, 440, 447, 491, 493, 494, 495, 504, 517], "squar": [0, 5, 6, 157, 188, 207, 209, 211, 220, 283, 303, 327, 340, 356, 416, 466, 468, 491, 492, 494, 495, 496, 520, 524], "squeez": [0, 435, 517], "src": [0, 135, 136], "ssh": [518, 522], "stabil": [155, 157, 361, 377, 381, 383, 416, 456, 457, 459, 491, 492, 493, 494, 495, 496, 504], "stabl": [227, 232, 300, 464], "stable_abi": 2, "stack": [0, 517], "standard": [0, 1, 4, 7, 52, 82, 233, 263, 267, 307, 434, 437, 439, 442, 518, 525], "starmap": [6, 340], "start": [0, 1, 2, 5, 6, 8, 9, 19, 158, 221, 240, 298, 299, 302, 342, 517, 519, 521, 522, 527], "start_axi": [0, 50, 175], "start_captur": 3, "start_indic": [298, 299], "state": [6, 7, 356, 376, 382, 417, 490, 501, 514, 517], "static": [9, 517], "static_cast": 2, "std": [0, 2, 4, 442, 519], "stderr": 522, "stdout": 522, "step": [0, 3, 4, 6, 7, 19, 335, 356, 376, 382, 417, 492, 499, 501, 506, 508, 509, 510, 517, 518], "step_decai": 490, "step_siz": 510, "still": [6, 9, 214, 517, 523], "stochast": [493, 494, 496, 505, 523], "stood": 6, "stop": [0, 2, 6, 19, 221, 241, 308, 520, 521], "stop_captur": 3, "stop_gradi": [0, 520], "storag": 87, "store": [6, 339], "str": [114, 126, 133, 140, 141, 150, 151, 152, 154, 156, 159, 180, 184, 190, 208, 210, 214, 222, 237, 238, 240, 258, 259, 284, 285, 286, 287, 288, 327, 336, 339, 343, 392, 393, 396, 397, 399, 401, 403, 409, 414, 415, 435, 439, 440, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 498], "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, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 124, 125, 126, 127, 128, 131, 132, 135, 136, 137, 138, 139, 140, 143, 144, 145, 147, 148, 149, 153, 155, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 177, 178, 179, 180, 185, 186, 187, 188, 189, 191, 192, 193, 194, 195, 196, 197, 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, 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, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 282, 283, 291, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 328, 329, 332, 333, 334, 335, 518, 527], "streamcontext": 309, "streamordevic": [0, 2], "street": 6, "strength": [497, 499, 505], "strict": [133, 185, 202, 397, 399, 409, 410, 411], "strictli": [214, 293], "stride": [0, 2, 87, 107, 108, 109, 110, 111, 112, 113, 358, 359, 360, 363, 364, 365, 366, 367, 368, 388, 389, 390, 421, 521], "string": [0, 2, 141, 150, 154, 156, 159, 190, 238, 253, 519, 524, 526], "stronger": 522, "structur": [2, 335, 500, 520], "stub": 9, "style": [2, 14, 17, 92, 94, 95, 138, 139, 143, 185, 186, 192, 201, 202, 203, 227, 233, 235, 243, 245, 249, 255, 276, 280, 310], "su": 6, "sub": [0, 7, 128, 271, 298, 299, 321, 336, 499], "subarrai": [128, 302], "subclass": 488, "subdivid": 1, "subdtyp": 198, "subgradi": 493, "sublinear": 492, "submodul": [6, 7, 356, 393, 397, 398, 409, 411], "subnetwork": 518, "suboptim": 519, "subscript": [140, 141], "subsect": 6, "subsequ": [133, 490, 518, 522], "subset": [356, 396, 410, 411], "substanti": 9, "subtl": 517, "subtract": [0, 39], "subtyp": [198, 347], "succe": 133, "successfulli": 518, "sudo": [9, 293, 518], "suggest": 518, "sum": [0, 2, 5, 14, 121, 132, 152, 191, 214, 232, 300, 318, 321, 356, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 518, 521, 524], "sum_": [214, 464], "sum_i": 455, "sum_j": [478, 479], "summat": [140, 141], "super": [6, 7, 356, 488], "superset": [340, 500], "support": [1, 2, 6, 8, 9, 17, 96, 109, 112, 113, 126, 159, 175, 187, 192, 204, 205, 207, 208, 209, 210, 211, 215, 216, 219, 220, 222, 233, 258, 266, 518, 520, 521, 524, 526], "suppos": [520, 527], "sure": [2, 3, 6, 9, 356, 517], "surpass": [439, 440], "surpris": 6, "sw": 1, "swap": [0, 114, 292, 312, 411], "swapax": [0, 122], "swiglu": 6, "swish": [424, 477], "switch": 9, "symbol": 497, "symmetr": [108, 109, 112, 113, 204, 205, 208, 210], "symmetri": [208, 210], "synchron": [2, 517], "syntax": [39, 521], "synthet": 5, "sysctl": 293, "system": [4, 6, 9, 181, 182, 217, 218, 238, 293], "t": [0, 1, 2, 4, 6, 9, 106, 144, 154, 156, 158, 159, 180, 204, 205, 259, 327, 356, 376, 382, 417, 491, 492, 493, 494, 495, 496, 497, 504, 505, 517, 519, 520, 527], "t_kv": 159, "t_q": 159, "tabl": [1, 214, 347, 373], "take": [0, 2, 6, 7, 92, 93, 94, 95, 102, 150, 179, 184, 199, 235, 243, 251, 259, 315, 327, 330, 331, 334, 341, 342, 358, 359, 360, 388, 389, 390, 412, 456, 498, 514, 518, 519, 520, 521, 522, 526, 527, 528], "take_along_axi": [0, 212, 521], "taken": [128, 314, 321], "talk": 518, "tan": 0, "tangent": [0, 2, 24, 25, 26, 122, 199, 316, 317, 433, 483], "tangent_i": 2, "tangent_x": 2, "tanh": [0, 356, 374, 376, 382, 391, 417, 447, 470], "target": [2, 327, 456, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 517], "target_include_directori": 2, "target_link_librari": [2, 4], "target_link_opt": 2, "target_sourc": 2, "task": 464, "tau": 505, "tcp": 518, "tediou": 518, "tell": [4, 6, 517, 524], "temp": 6, "templat": [0, 1, 2, 154, 156], "ten": 523, "tend": 497, "tensor": [222, 318, 469, 524], "tensordot": 0, "term": [2, 459, 491, 492, 493, 494, 495, 496, 504], "termin": [9, 522], "test": [7, 9, 518, 522], "test_imag": 7, "test_label": 7, "text": [6, 267, 374, 376, 382, 391, 417, 425, 432, 437, 438, 439, 440, 447, 450, 451, 452, 459, 460, 461, 464, 465, 468, 470, 471, 475, 476, 481, 482, 492, 497], "textrm": [258, 374, 375, 446, 449], "tf": 524, "tgp_size": 2, "th": [118, 119, 120, 121, 127, 153, 207, 208, 228, 508], "than": [1, 2, 6, 82, 114, 128, 139, 158, 179, 185, 186, 202, 203, 204, 205, 207, 208, 209, 210, 211, 212, 215, 219, 220, 233, 289, 293, 338, 340, 421, 432, 435, 465, 468, 482, 492, 497, 498, 517, 519, 520, 527], "thank": 523, "thei": [1, 2, 5, 6, 9, 17, 114, 180, 192, 258, 423, 460, 488, 497, 516, 517, 518, 519, 523, 525, 526, 527], "them": [0, 2, 6, 131, 356, 397, 409, 518, 519, 522, 527], "themselv": [2, 517], "thi": [0, 1, 2, 4, 6, 7, 9, 16, 17, 18, 19, 27, 28, 29, 30, 87, 88, 100, 122, 142, 150, 152, 154, 156, 179, 180, 181, 187, 190, 192, 199, 204, 205, 207, 208, 209, 210, 211, 214, 215, 216, 219, 220, 227, 232, 233, 234, 236, 242, 254, 256, 262, 291, 293, 300, 301, 302, 307, 311, 314, 320, 328, 335, 338, 341, 342, 356, 369, 370, 371, 375, 376, 382, 393, 394, 396, 397, 400, 401, 402, 407, 409, 410, 411, 412, 415, 417, 432, 437, 438, 439, 440, 447, 448, 449, 456, 464, 482, 488, 498, 501, 516, 517, 518, 519, 520, 522, 523, 524, 526], "thin": 522, "thing": [2, 6], "third": [206, 360, 390, 518, 519], "this_grid": 154, "thompson": 370, "those": [2, 6, 356, 499], "though": [2, 6, 517, 519, 523, 524], "thousand": 523, "thread": [1, 2], "thread_index_in_simdgroup": 1, "thread_position_in_grid": [1, 2, 156], "thread_rank": 154, "threadgroup": [1, 2, 154, 156], "threads_per_simdgroup": 1, "three": [6, 91, 258, 360, 390, 435], "threefri": 514, "threshold": [432, 461, 468, 482], "through": [1, 2, 308, 434, 497, 517, 518, 519, 520, 524], "throw": [2, 102, 133], "thu": [6, 356], "thumb": 490, "tic": 517, "tieleman": 504, "tile": [0, 159], "time": [1, 2, 6, 9, 292, 319, 356, 376, 382, 417, 517, 518, 520, 523, 527], "timeit": [517, 520], "titl": 2, "tmp": [1, 154, 156], "to_quant": 336, "to_stream": 2, "toc": 517, "togeth": [0, 1, 2, 7, 258, 340, 341, 518], "tok_embed": 6, "token": [6, 373, 414], "told": 6, "toler": [0, 17, 192], "too": [198, 517, 523], "took": 6, "tool": 9, "toolkit": 9, "top": [2, 320, 385, 435], "topk": 0, "torch": [6, 524], "torch_weight": 6, "total": [293, 520], "total_norm": 338, "tpi": 517, "tpng": 518, "trace": [0, 3, 152, 517], "trace_fil": 3, "tracer": 410, "track": [2, 356, 361], "track_running_stat": 361, "trade": 523, "tradit": [6, 158, 370, 371, 421], "train": [6, 7, 356, 361, 369, 370, 371, 395, 397, 409, 437, 438, 519], "train_imag": [7, 490], "train_label": [7, 490], "trainabl": [7, 337, 356, 488], "trainable_paramet": [356, 396, 501], "transfer": 522, "transform": [1, 6, 8, 122, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 187, 337, 356, 361, 377, 383, 385, 396, 397, 409, 415, 421, 521], "transformerencod": 287, "transit": 508, "translat": [155, 383], "transpos": [0, 6, 32, 111, 112, 113, 180, 259, 366, 367, 368], "treat": [0, 2, 122, 169, 170, 172, 173, 314, 435, 517], "tree": [8, 88, 102, 146, 184, 327, 331, 335, 339, 340, 341, 342, 343, 500, 501, 503, 512, 520], "tree_flatten": [287, 340, 343, 356, 490, 519], "tree_map": [341, 356, 518], "tree_unflatten": [6, 490, 519], "trembl": 6, "tri": [0, 133], "triangl": [208, 210, 323], "triangular": [204, 205, 218, 220], "trigger": 517, "tril": 0, "trilinear": 435, "triplet": 469, "triplet_loss": 356, "triu": 0, "true": [0, 1, 2, 4, 5, 6, 17, 42, 43, 44, 45, 58, 86, 102, 118, 119, 120, 121, 154, 156, 158, 180, 192, 198, 204, 205, 214, 219, 222, 228, 237, 259, 300, 332, 336, 339, 340, 341, 342, 347, 356, 361, 363, 364, 365, 366, 367, 368, 376, 377, 381, 382, 383, 385, 396, 397, 399, 406, 409, 410, 411, 415, 417, 421, 426, 434, 435, 456, 464, 492, 494, 495, 498, 499, 517, 519], "truncat": [160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 272], "truth": [5, 458, 468], "try": [2, 9, 518], "tupl": [0, 31, 69, 72, 75, 83, 88, 97, 103, 108, 109, 110, 112, 113, 135, 139, 141, 146, 148, 150, 161, 162, 165, 166, 169, 170, 172, 173, 190, 199, 207, 208, 212, 213, 214, 216, 219, 253, 258, 279, 281, 298, 299, 305, 326, 327, 330, 339, 340, 341, 342, 343, 358, 359, 360, 364, 365, 367, 368, 388, 389, 390, 399, 401, 423, 435, 492, 494, 495, 496, 497, 516, 519, 520], "tutori": 2, "twice": 527, "two": [0, 2, 14, 15, 17, 25, 86, 90, 92, 94, 95, 96, 128, 138, 143, 161, 165, 172, 179, 180, 185, 186, 192, 200, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 215, 216, 219, 220, 227, 233, 235, 243, 245, 249, 252, 258, 312, 342, 359, 375, 382, 389, 449, 457, 517, 518, 519, 520, 521, 527], "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, 72, 82, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 125, 126, 127, 128, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 143, 144, 145, 147, 148, 149, 153, 155, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 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, 242, 243, 244, 245, 246, 247, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 277, 279, 280, 282, 283, 289, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 338, 339, 342, 356, 404, 434, 436, 437, 438, 439, 440, 441, 442, 443, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 517, 519, 521, 524], "type_to_nam": 2, "typenam": [0, 1, 2], "typic": [0, 159, 335, 373, 490, 517, 523], "u": [1, 2, 4, 204, 208, 210, 212, 219, 385, 411, 512, 518, 522, 523], "u_": 491, "u_t": 491, "ubuntu": 9, "ubuntu2204": 9, "uint": [1, 2, 156], "uint16": [12, 347], "uint3": 1, "uint32": [12, 27, 28, 29, 30, 262, 347], "uint64": [12, 347], "uint8": [12, 347], "ultra": 6, "unabl": 9, "unam": 9, "unari": 517, "unchang": [158, 308, 421], "uncheck": 9, "uncompress": 287, "undefin": [0, 29, 122, 204, 205, 254, 266, 521], "under": 214, "underli": [2, 329], "understand": [6, 437, 438], "unevalu": 151, "unexpect": [2, 19], "unexpectedli": 522, "unflatten": 0, "unfreez": [356, 397], "unfrozen": 409, "unifi": 8, "uniform": [3, 356, 385, 399, 438, 440, 485, 514, 517, 520, 527], "uniformli": 273, "unintend": 0, "union": [19, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 89, 90, 91, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 196, 197, 198, 200, 209, 210, 219, 238, 285, 309, 339], "uniqu": [2, 217, 218, 514, 518], "unique_ptr": 2, "unit": [362, 372, 374, 375, 376, 384, 418, 420, 422, 424, 437, 438, 439, 440, 444, 445, 446, 447, 448, 449, 453, 472, 474, 475, 477], "unittest": 9, "univers": 214, "unless": [6, 17, 192, 214, 488], "unlik": [6, 17, 192, 212, 258, 370, 371, 405, 521], "unnecessari": [2, 6], "unnorm": [262, 456, 458], "unscal": 492, "unsign": [180, 258, 259, 347], "unsignedinteg": 12, "unspecifi": [16, 18, 19, 27, 28, 29, 30, 103, 118, 119, 120, 121, 178, 228, 232, 234, 236, 242, 250, 254, 256, 277, 300, 301, 307, 311, 314, 320, 321, 328, 333, 335, 528], "unsqueez": 6, "unsupport": 222, "until": [2, 335, 523, 525], "unus": 2, "up": [1, 2, 6, 122, 517], "upcast": 2, "updat": [0, 1, 2, 5, 6, 7, 9, 39, 102, 299, 336, 340, 342, 361, 392, 393, 399, 404, 405, 406, 411, 490, 492, 495, 497, 500, 501, 505, 506, 507, 508, 509, 510, 517, 518, 519, 523], "update_modul": 356, "uplo": [208, 210], "upon": [6, 340, 341], "upper": [204, 205, 208, 210, 218, 220, 258, 269, 272, 273, 443], "upsampl": 356, "us": [0, 3, 5, 6, 7, 8, 9, 19, 39, 87, 122, 126, 129, 131, 132, 135, 136, 137, 139, 154, 156, 158, 175, 180, 181, 182, 183, 201, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 233, 258, 259, 267, 277, 279, 280, 281, 289, 292, 293, 313, 326, 335, 336, 339, 342, 347, 349, 356, 359, 360, 370, 373, 374, 376, 382, 385, 389, 390, 392, 396, 403, 410, 412, 414, 415, 417, 421, 426, 434, 435, 439, 440, 447, 448, 457, 485, 488, 490, 491, 492, 494, 495, 496, 497, 498, 500, 501, 514, 516, 517, 518, 519, 520, 521, 522, 525, 527], "usag": [434, 517, 518], "user": [2, 6, 356], "uss": 335, "usual": [373, 414, 516, 523], "util": [1, 2, 6, 8, 9, 287, 356, 490, 522], "v": [6, 114, 159, 207, 208, 356, 397, 524], "v_": [491, 493, 494, 495, 496, 504, 505], "v_t": [491, 493, 494, 495, 496, 504, 505], "val": [0, 31, 178], "valid": [7, 114, 175, 331, 339, 397, 409, 516, 518], "valid_parameter_filt": 392, "valu": [0, 1, 5, 6, 12, 13, 17, 19, 27, 28, 52, 82, 86, 101, 133, 150, 153, 154, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 175, 178, 190, 192, 206, 214, 215, 219, 221, 238, 246, 253, 257, 258, 261, 262, 263, 265, 266, 267, 269, 272, 273, 281, 285, 293, 314, 315, 327, 331, 337, 339, 340, 341, 342, 347, 359, 360, 362, 369, 370, 371, 372, 378, 381, 385, 389, 390, 396, 412, 413, 430, 432, 434, 436, 456, 457, 458, 459, 460, 461, 463, 464, 465, 466, 467, 468, 482, 488, 492, 495, 506, 507, 509, 510, 520], "value_and_grad": [7, 122, 356, 410, 488, 490, 503, 517, 520, 524, 525], "value_and_grad_fn": 523, "value_cach": 6, "value_dim": 412, "value_input_dim": 412, "value_output_dim": 412, "value_proj": 6, "valueerror": [122, 214, 399, 520], "values_hat": 6, "van": 214, "var": [0, 361, 377, 381, 383, 459], "vari": 335, "variabl": [9, 102, 122, 129, 142, 150, 151, 152, 184, 199, 327, 330, 331, 517, 518, 519], "varianc": [0, 307, 328, 361, 377, 459], "variant": [6, 468, 496], "variou": 214, "vector": [0, 2, 5, 8, 158, 191, 199, 214, 314, 330, 331, 373, 458, 519, 525], "verbos": [1, 154, 156, 518], "veri": [6, 412, 522, 523, 527], "verifi": [5, 9], "versa": 281, "version": [2, 4, 9, 88, 126, 150, 152, 190, 227, 232, 258, 300, 331, 514, 520, 521], "versu": 517, "via": [9, 122, 500, 503, 518, 522, 523, 524], "vice": 281, "video": 371, "view": [0, 3, 87, 521, 524], "visual": 151, "vjp": [2, 122, 525], "vmap": [2, 122, 519, 520, 523, 525], "vmap_add": 520, "vocab_s": 6, "vocabulari": [373, 414], "void": [1, 2], "volta": 9, "vt": 219, "w": [0, 1, 5, 108, 109, 112, 113, 126, 180, 207, 208, 258, 259, 327, 341, 361, 364, 365, 367, 368, 370, 371, 385, 490, 505, 520], "w1": [6, 338], "w2": [6, 338], "w3": 6, "w_": [376, 382, 417, 491, 492, 493, 494, 495, 496, 497, 504, 505], "w_i": [126, 258], "w_in": 1, "w_q": 258, "w_star": 5, "w_stride": 1, "w_t": [491, 493, 494, 495, 496, 497, 504, 505], "wa": [4, 6, 87, 135, 136, 518, 519, 523], "wai": [2, 6, 9, 356, 435, 517, 518, 519, 520, 521, 522], "wait": 6, "walk": [6, 519], "walkthrough": 2, "walsh": 187, "want": [1, 2, 6, 518, 519, 520, 522, 527], "warm": [2, 517], "warmup_init": 492, "watch": [6, 517], "wd": 497, "we": [0, 1, 2, 5, 6, 7, 122, 126, 135, 136, 180, 258, 259, 356, 373, 414, 423, 495, 497, 514, 516, 517, 518, 519, 520, 522, 523, 527], "weight": [0, 5, 107, 108, 109, 110, 111, 112, 113, 155, 157, 340, 356, 399, 403, 414, 415, 456, 458, 488, 492, 495, 497, 498, 499, 501, 505, 520, 523], "weight_decai": [492, 495, 497, 499, 505], "weight_fil": 6, "weights_fp16": 523, "well": [6, 356, 397, 409, 412, 518, 523], "wen": 6, "went": 6, "were": [6, 527], "wet": 6, "wget": 9, "what": [2, 6, 340, 522], "whatsoev": 6, "whc": 370, "when": [0, 1, 2, 6, 8, 9, 102, 110, 122, 137, 204, 205, 207, 208, 209, 210, 211, 214, 215, 219, 220, 222, 292, 363, 364, 365, 366, 367, 368, 435, 439, 440, 456, 462, 468, 488, 490, 508, 514, 517, 518, 519, 527], "where": [0, 4, 7, 153, 158, 192, 205, 258, 327, 331, 361, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 376, 377, 381, 382, 383, 385, 396, 413, 416, 417, 432, 439, 440, 445, 446, 448, 459, 465, 471, 475, 477, 482, 501, 518, 520, 521], "wherea": 520, "whether": [150, 152, 154, 156, 180, 208, 210, 218, 220, 259, 376, 382, 396, 412, 417, 456, 459, 465], "which": [0, 1, 2, 6, 7, 8, 9, 19, 38, 87, 88, 102, 110, 128, 131, 132, 133, 135, 136, 137, 146, 150, 152, 158, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 184, 190, 193, 194, 195, 196, 197, 199, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 222, 237, 240, 258, 262, 263, 277, 279, 281, 284, 285, 286, 287, 288, 305, 306, 314, 321, 326, 327, 330, 331, 335, 336, 359, 360, 370, 371, 374, 389, 390, 392, 396, 421, 456, 458, 461, 465, 468, 485, 500, 501, 514, 517, 518, 519, 520, 521, 522, 523, 527, 528], "while": [2, 3, 6, 9, 167, 279, 421, 523, 524], "whistl": 2, "who": 6, "whose": [153, 336, 337], "why": 6, "wi": 518, "wide": 523, "width": [359, 360, 361, 364, 365, 367, 368, 370, 371, 389, 390, 414, 415], "window": [9, 358, 359, 360, 388, 389, 390], "wipe": 9, "wire": 293, "wired_limit_mb": 293, "wise": [0, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 92, 93, 94, 95, 99, 115, 116, 138, 139, 143, 144, 145, 147, 149, 176, 177, 185, 186, 192, 201, 202, 203, 223, 224, 225, 226, 227, 229, 230, 231, 235, 243, 245, 247, 249, 255, 275, 276, 280, 283, 294, 295, 296, 297, 303, 304, 310, 316, 317, 362, 370, 371, 380, 391, 413, 425, 444, 451, 452, 454, 455, 470, 471, 473, 474, 477, 478, 479, 480, 517], "wish": 9, "with_logit": 456, "within": [0, 3, 29, 192], "without": [1, 6, 8, 308, 412, 484, 516, 517, 518, 519, 522, 523, 524, 527], "wk": 6, "wl": 2, "wo": 6, "word": 0, "work": [2, 3, 6, 292, 347, 499, 517, 518, 519, 520, 521, 522, 523], "workhors": 356, "world": [343, 518], "world2": 518, "world_ani": 518, "world_mpi": 518, "world_r": 518, "worri": [1, 523], "would": [2, 6, 435, 518, 519, 521, 523, 524, 527], "wq": 6, "wrap": [122, 356, 498], "wrapper": [519, 522], "write": [0, 1, 6, 356, 524], "written": [2, 519], "wrong": 519, "wrt": 337, "wv": 6, "x": [0, 1, 2, 4, 5, 6, 7, 39, 88, 96, 122, 131, 132, 136, 137, 144, 149, 150, 151, 155, 156, 157, 180, 187, 188, 190, 214, 259, 263, 267, 268, 282, 287, 294, 324, 325, 332, 340, 342, 356, 358, 359, 360, 361, 362, 372, 374, 375, 377, 381, 383, 384, 385, 388, 389, 390, 391, 392, 413, 416, 418, 425, 426, 432, 435, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 468, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 488, 490, 497, 517, 518, 519, 520, 521, 523, 524, 525, 527], "x1": 457, "x2": 457, "x86_64": 9, "x_1": [457, 465], "x_2": [457, 465], "x_cast": 2, "x_grad": 1, "x_i": [455, 478, 479], "x_j": [478, 479], "x_offset": 2, "x_ptr": 2, "x_shape": 1, "x_stride": 2, "x_t": [376, 382, 417], "x_view": 524, "xcode": 9, "xcodeproj": 3, "xcrun": 9, "xf": 382, "xg": 382, "xi": 382, "xn": 376, "xo": 382, "xor": 95, "xr": 376, "xy": [0, 237], "xz": 376, "x\u00b2": 524, "y": [0, 2, 4, 5, 6, 7, 9, 39, 88, 122, 150, 151, 187, 190, 332, 356, 361, 370, 377, 381, 383, 385, 416, 460, 465, 468, 490, 493, 517, 518, 519, 520, 523, 524], "y_": [460, 464], "y_cast": 2, "y_hat": 356, "y_offset": 2, "y_ptr": 2, "y_stride": 2, "ye": 6, "year": 6, "yet": [6, 356, 488, 501, 520, 521, 523, 525], "yield": [6, 7, 514], "you": [1, 2, 3, 4, 6, 7, 8, 9, 293, 356, 426, 434, 485, 514, 517, 518, 519, 520, 521, 522, 524, 526, 527], "your": [2, 6, 9, 488, 518, 520, 523], "z": [2, 88, 376, 517, 519, 523], "z_t": 376, "zeiler": 491, "zero": [0, 150, 153, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 171, 172, 173, 206, 237, 278, 299, 323, 324, 325, 334, 356, 358, 359, 360, 369, 370, 371, 399, 436, 437, 438, 439, 440, 441, 442, 443, 485, 490, 492, 519, 521], "zero_grad": 520, "zeros_lik": [0, 212], "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.imag", "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.logcumsumexp", "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.real", "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.cuda.is_available", "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.cuda_kernel", "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.fftshift", "mlx.core.fft.ifft", "mlx.core.fft.ifft2", "mlx.core.fft.ifftn", "mlx.core.fft.ifftshift", "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.eig", "mlx.core.linalg.eigh", "mlx.core.linalg.eigvals", "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.logcumsumexp", "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", "CUDA", "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.ReLU2", "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.relu2", "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.MultiOptimizer", "mlx.optimizers.Muon", "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": 527, "In": 521, "The": 356, "ab": [13, 33], "adadelta": 491, "adafactor": 492, "adagrad": 493, "adam": 494, "adamax": 496, "adamw": 495, "add": 14, "addmm": 15, "algebra": 353, "alibi": 357, "all": [6, 16, 34, 518], "all_gath": 131, "all_sum": 132, "allclos": 17, "ani": [18, 35], "api": [8, 9], "appli": 392, "apply_gradi": 500, "apply_to_modul": 393, "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, 83, 84, 85, 345, 521, 526], "array_equ": 86, "as_strid": 87, "astyp": 38, "async_ev": 88, "atleast_1d": 89, "atleast_2d": 90, "atleast_3d": 91, "attent": 6, "automat": 520, "average_gradi": [335, 518], "avgpool1d": 358, "avgpool2d": 359, "avgpool3d": 360, "back": 2, "backend": 518, "basic": [517, 519, 525], "batchnorm": 361, "benchmark": 6, "bernoulli": 261, "binari": 9, "binary_cross_entropi": 456, "bind": 2, "bitwise_and": 92, "bitwise_invert": 93, "bitwise_or": 94, "bitwise_xor": 95, "block_masked_mm": 96, "broadcast_arrai": 97, "broadcast_to": 98, "build": [2, 9], "c": [4, 8, 9, 519], "categor": 262, "ceil": 99, "celu": [362, 444], "children": 394, "choleski": 204, "cholesky_inv": 205, "class": 356, "clear_cach": 100, "clip": 101, "clip_grad_norm": 338, "cmake": 2, "co": [41, 115], "code": [2, 6], "common": 511, "commun": [349, 518], "compil": [102, 517], "complex": 1, "comput": 523, "concaten": 103, "conj": [40, 104], "conjug": 105, "constant": 436, "contigu": 106, "conv1d": [107, 363], "conv2d": [108, 364], "conv3d": [109, 365], "conv_gener": 110, "conv_transpose1d": 111, "conv_transpose2d": 112, "conv_transpose3d": 113, "convers": 524, "convert": 6, "convolv": 114, "convtranspose1d": 366, "convtranspose2d": 367, "convtranspose3d": 368, "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, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 344], "cosh": 116, "cosine_decai": 506, "cosine_similarity_loss": 457, "cpu": [2, 9], "cross": 206, "cross_entropi": 458, "cuda": [9, 117, 346], "cuda_kernel": 154, "cummax": [42, 118], "cummin": [43, 119], "cumprod": [44, 120], "cumsum": [45, 121], "custom": [1, 2], "custom_funct": 122, "data": 347, "debug": 517, "debugg": 3, "default_devic": 123, "default_stream": 124, "defin": 518, "degre": 125, "dequant": 126, "devic": [10, 348], "device_info": 238, "diag": [46, 127], "diagon": [47, 128], "differ": 521, "differenti": 520, "disable_compil": 129, "distribut": [130, 131, 132, 133, 134, 135, 136, 137, 349, 518, 522], "divid": 138, "divmod": 139, "download": [2, 6], "dropout": 369, "dropout2d": 370, "dropout3d": 371, "dtype": [11, 48], "dtypecategori": 12, "eig": 207, "eigh": 208, "eigval": 209, "eigvalsh": 210, "einsum": 140, "einsum_path": 141, "elu": [372, 445], "embed": 373, "enable_compil": 142, "encod": 6, "end": 2, "equal": 143, "erf": 144, "erfinv": 145, "eval": [146, 395], "evalu": 523, "exampl": [1, 2, 8, 517, 518, 519, 527], "exp": [49, 147], "expand_dim": 148, "expm1": 149, "exponential_decai": 507, "export": [152, 350, 519], "export_funct": 150, "export_to_dot": 151, "extens": 2, "ey": 153, "fast": [154, 155, 156, 157, 158, 159, 351], "fft": [160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 352], "fft2": 161, "fftn": 162, "fftshift": 163, "filter_and_map": 396, "finfo": 174, "flatten": [50, 175], "floor": 176, "floor_divid": 177, "format": 526, "found": 9, "framework": 524, "freez": 397, "from": [9, 521], "full": [6, 178], "function": [350, 484, 487, 517, 519, 520, 525], "further": 8, "gather_mm": 179, "gather_qmm": 180, "gaussian_nll_loss": 459, "gelu": [374, 446], "gelu_approx": 447, "gelu_fast_approx": 448, "gener": 6, "get": 518, "get_active_memori": 181, "get_cache_memori": 182, "get_peak_memori": 183, "glorot_norm": 437, "glorot_uniform": 438, "glu": [375, 449], "gpu": 2, "grad": [184, 356], "graph": [517, 523, 525], "greater": 185, "greater_equ": 186, "grid": 1, "group": 130, "groupnorm": 377, "gru": 376, "guid": 525, "gumbel": 263, "hadamard_transform": 187, "hard_shrink": 450, "hard_tanh": 451, "hardshrink": 378, "hardswish": [380, 452], "hardtanh": 379, "he_norm": 439, "he_uniform": 440, "hinge_loss": 460, "host": [518, 522], "huber_loss": 461, "ident": [188, 441], "ifft": 164, "ifft2": 165, "ifftn": 166, "ifftshift": 167, "imag": [51, 189], "implement": [2, 6], "import": 519, "import_funct": 190, "index": 521, "infer": 6, "init": [133, 436, 437, 438, 439, 440, 441, 442, 443, 501], "initi": 485, "inner": 191, "inspect": 356, "instal": [8, 9, 518], "instancenorm": 381, "introduc": 2, "inv": 211, "irfft": 168, "irfft2": 169, "irfftn": 170, "is_avail": [117, 134, 239], "isclos": 192, "isfinit": 193, "isinf": 194, "isnan": 195, "isneginf": 196, "isposinf": 197, "issubdtyp": 198, "item": 52, "items": 53, "jax": 524, "join_schedul": 508, "jvp": 199, "kei": 264, "kernel": 1, "kl_div_loss": 462, "kron": 200, "l1_loss": 463, "laplac": 265, "launch": 522, "layer": [6, 7, 486], "layer_norm": 155, "layernorm": 383, "lazi": 523, "leaf_modul": 398, "leaky_relu": 453, "leakyrelu": 384, "left_shift": 201, "less": 202, "less_equ": 203, "linalg": [204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220], "linear": [5, 353, 385], "linear_schedul": 509, "linspac": 221, "linux": 9, "lion": 497, "llm": 6, "load": [6, 222, 490, 526], "load_weight": 399, "log": [54, 223], "log10": [55, 224], "log1p": [56, 225], "log2": [57, 226], "log_cosh_loss": 464, "log_sigmoid": 454, "log_softmax": 455, "logaddexp": 227, "logcumsumexp": [58, 228], "logical_and": 229, "logical_not": 230, "logical_or": 231, "logsigmoid": 386, "logsoftmax": 387, "logsumexp": [59, 232], "loss": [456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 487], "lstm": 382, "lu": 212, "lu_factor": 213, "manag": 354, "margin_ranking_loss": 465, "matmul": 233, "max": [60, 234], "maximum": 235, "maxpool1d": 388, "maxpool2d": 389, "maxpool3d": 390, "mean": [61, 236], "memori": [354, 527], "meshgrid": 237, "metal": [1, 3, 9, 238, 239, 240, 241, 355], "metal_kernel": 156, "min": [62, 242], "minim": 9, "minimum": 243, "mish": [391, 470], "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, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 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, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510], "model": 6, "modul": [356, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 488, 519], "more": 519, "moveaxi": [63, 244], "mpi": [518, 522], "mse_loss": 466, "multi": 7, "multiheadattent": 412, "multioptim": 498, "multipl": 519, "multipli": 245, "multivariate_norm": 266, "muon": 499, "named_modul": 401, "nan_to_num": 246, "nbyte": 64, "ndim": 65, "neg": 247, "network": 356, "neural": 356, "new_stream": 248, "nll_loss": 467, "nn": [335, 336, 337, 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, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 518], "norm": 214, "normal": [267, 442], "not_equ": 249, "numpi": [521, 524], "ones": 250, "ones_lik": 251, "onli": [9, 523], "oper": [0, 2, 489], "optim": [338, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512], "option": 9, "other": 524, "outer": 252, "packag": 4, "pad": 253, "paramet": [356, 402], "partit": 254, "perceptron": 7, "permut": 268, "pinv": 215, "place": 521, "power": 255, "prelu": [413, 471], "primit": 2, "prod": [66, 256], "program": [518, 522], "provid": 522, "pure": 517, "put": 6, "put_along_axi": 257, "python": [2, 8, 9], "pytorch": 524, "qr": 216, "quantiz": [258, 336], "quantized_matmul": 259, "quantizedembed": 414, "quantizedlinear": 415, "quick": [356, 525], "radian": 260, "randint": 269, "random": [261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 514], "read": 8, "real": [67, 274], "reciproc": [68, 275], "recv": 135, "recv_lik": 136, "reduc": 518, "refer": 8, "regress": 5, "relu": [418, 472], "relu2": [419, 473], "relu6": [420, 474], "remaind": 276, "remot": [518, 522], "repeat": 277, "requir": 9, "reset_peak_memori": 278, "reshap": [69, 279], "result": 2, "rfft": 171, "rfft2": 172, "rfftn": 173, "right_shift": 280, "ring": [518, 522], "rms_norm": 157, "rmsnorm": 416, "rmsprop": 504, "rnn": 417, "roll": 281, "rope": [158, 421], "round": [70, 282], "rsqrt": [71, 283], "run": 518, "sampl": 1, "save": [284, 490, 526], "save_gguf": 285, "save_safetensor": 286, "save_weight": 403, "savez": 287, "savez_compress": 288, "scaled_dot_product_attent": 159, "schedul": 513, "script": [2, 6], "seed": 270, "select": 518, "selu": [422, 475], "send": 137, "sequenti": 423, "serial": 526, "set": [518, 522], "set_cache_limit": 289, "set_default_devic": 290, "set_default_stream": 291, "set_dtyp": 404, "set_memory_limit": 292, "set_wired_limit": 293, "setuptool": 2, "sgd": 505, "shape": [1, 72], "shapeless": [517, 519], "shell": 9, "sigmoid": [294, 425, 476], "sign": 295, "silu": [424, 477], "simpl": [1, 527], "sin": [73, 296], "sinh": 297, "sinusoidalpositionalencod": 426, "size": [9, 74], "slice": 298, "slice_upd": 299, "smooth_l1_loss": 468, "softmax": [300, 427, 478], "softmin": [428, 479], "softplu": [429, 480], "softshrink": [430, 481], "softsign": 431, "solv": 217, "solve_triangular": 218, "sort": 301, "sourc": 9, "specif": 522, "specifi": 528, "speedup": 517, "split": [75, 271, 302], "sqrt": [76, 303], "squar": [77, 304], "squeez": [78, 305], "stack": 306, "start": [356, 518, 525], "start_captur": 240, "state": [405, 502], "std": [79, 307], "step": [432, 482], "step_decai": 510, "stop_captur": 241, "stop_gradi": 308, "stream": [309, 344, 348, 528], "stride": 1, "subtract": 310, "sum": [80, 311], "support": 347, "svd": 219, "swapax": [81, 312], "synchron": 313, "t": 32, "take": 314, "take_along_axi": 315, "tan": 316, "tanh": [317, 433, 483], "tensordot": 318, "tensorflow": 524, "thunderbolt": 518, "tile": 319, "togeth": 6, "tolist": 82, "topk": 320, "trace": [321, 519], "train": [406, 408, 517, 518], "trainable_paramet": 407, "transform": [2, 434, 515, 517, 519, 520, 523, 525], "transpos": [83, 322], "tree": 516, "tree_flatten": 339, "tree_map": 340, "tree_map_with_path": 341, "tree_reduc": 342, "tree_unflatten": 343, "tri": 323, "tri_inv": 220, "tril": 324, "triplet_loss": 469, "triu": 325, "troubleshoot": 9, "truncated_norm": 272, "tune": 518, "type": 347, "unflatten": 326, "unfreez": 409, "unifi": 527, "uniform": [273, 443], "up": [518, 522], "updat": [356, 410, 503, 521], "update_modul": 411, "upsampl": 435, "us": [1, 2, 4, 523, 528], "usag": [2, 8, 522], "util": [339, 340, 341, 342, 343, 516, 518], "valu": 356, "value_and_grad": [327, 337], "var": [84, 328], "variabl": 4, "vector": 520, "view": [85, 329], "vjp": [1, 330], "vmap": 331, "weight": 6, "what": 523, "when": 523, "where": 332, "why": 523, "workflow": 3, "x86": 9, "xcode": 3, "you": 523, "zero": 333, "zeros_lik": 334}}) \ No newline at end of file +Search.setIndex({"alltitles": {"A Simple Example": [[527, "a-simple-example"]], "Array": [[345, null]], "Attention layer": [[6, "attention-layer"]], "Automatic Differentiation": [[520, "automatic-differentiation"]], "Automatic Vectorization": [[520, "automatic-vectorization"]], "Basics": [[525, "basics"]], "Basics of Compile": [[517, "basics-of-compile"]], "Basics of Exporting": [[519, "basics-of-exporting"]], "Binary Size Minimization": [[9, "binary-size-minimization"]], "Binding to Python": [[2, "binding-to-python"]], "Build Options": [[9, "id5"]], "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]], "CPU-only (Linux)": [[9, "cpu-only-linux"]], "CUDA": [[9, "cuda"], [9, "id3"], [346, null]], "Common Optimizers": [[511, null]], "Compilation": [[517, null]], "Compiling Training Graphs": [[517, "compiling-training-graphs"]], "Complex Example": [[1, "complex-example"]], "Conversion to NumPy and Other Frameworks": [[524, null]], "Converting the weights": [[6, "converting-the-weights"]], "Custom Extensions in MLX": [[2, null]], "Custom Metal Kernels": [[1, null]], "Data Types": [[347, null]], "Debugging": [[517, "debugging"]], "Defining a Ring": [[518, "defining-a-ring"]], "Devices and Streams": [[348, null]], "Differences from NumPy": [[521, "differences-from-numpy"]], "Distributed Communication": [[349, null], [518, null]], "Download the code": [[2, null], [6, null]], "Encoder layer": [[6, "encoder-layer"]], "Example Speedup": [[517, "example-speedup"]], "Examples": [[8, null]], "Export Functions": [[350, null]], "Exporting Functions": [[519, null]], "Exporting Modules": [[519, "exporting-modules"]], "Exporting Multiple Traces": [[519, "exporting-multiple-traces"]], "FFT": [[352, null]], "Fast": [[351, null]], "Full model": [[6, "full-model"]], "Function Transforms": [[520, null]], "Function and Graph Transformations": [[525, "function-and-graph-transformations"]], "Functions": [[484, null]], "Further Reading": [[8, null]], "Generation": [[6, "generation"]], "Getting Started": [[518, "getting-started"]], "Getting Started with MPI": [[518, "getting-started-with-mpi"]], "Getting Started with Ring": [[518, "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++": [[519, "importing-functions-in-c"]], "In Place Updates": [[521, "in-place-updates"]], "Indexing Arrays": [[521, null]], "Initializers": [[485, null]], "Inspecting Modules": [[356, "inspecting-modules"]], "Install": [[8, null]], "Installing MPI": [[518, "installing-mpi"]], "Introducing the Example": [[2, "introducing-the-example"]], "JAX": [[524, "jax"]], "LLM inference": [[6, null]], "Launching Distributed Programs": [[522, null]], "Layers": [[486, null]], "Lazy Evaluation": [[523, null]], "Linear Algebra": [[353, null]], "Linear Regression": [[5, null]], "Linux": [[9, "linux"]], "Loss Functions": [[487, null]], "MLX": [[8, null]], "MPI Specifics": [[522, "mpi-specifics"]], "Memory Management": [[354, null]], "Metal": [[355, null]], "Metal Debugger": [[3, null]], "Metal not found": [[9, "metal-not-found"]], "Module": [[488, null]], "More Examples": [[519, "more-examples"]], "Multi-Layer Perceptron": [[7, null]], "Neural Networks": [[356, null]], "Only Compute What You Use": [[523, "only-compute-what-you-use"]], "Operations": [[0, null], [2, "operations"], [489, null]], "Operations and Primitives": [[2, "operations-and-primitives"]], "Optimizer": [[512, null]], "Optimizers": [[490, null]], "Package Variables": [[4, "id1"]], "Parameters": [[356, "parameters"]], "Primitive Transforms": [[2, "primitive-transforms"]], "Primitives": [[2, "primitives"]], "Providing Hosts": [[522, "providing-hosts"]], "Pure Functions": [[517, "pure-functions"]], "Putting it all together": [[6, "putting-it-all-together"]], "PyTorch": [[524, "pytorch"]], "Python API": [[9, "python-api"]], "Python API Reference": [[8, null]], "Python Installation": [[9, "python-installation"]], "Quick Start Guide": [[525, null]], "Quick Start with Neural Networks": [[356, "quick-start-with-neural-networks"]], "Random": [[514, null]], "Results": [[2, "results"]], "Ring Specifics": [[522, "ring-specifics"]], "Running Distributed Programs": [[518, "running-distributed-programs"]], "Saving and Loading": [[490, "saving-and-loading"]], "Saving and Loading Arrays": [[526, null]], "Schedulers": [[513, null]], "Scripts": [[2, "scripts"], [6, "scripts"]], "Selecting Backend": [[518, "selecting-backend"]], "Serialization Formats": [[526, "id1"]], "Setting up Remote Hosts": [[518, "setting-up-remote-hosts"], [522, "setting-up-remote-hosts"]], "Shapeless Compilation": [[517, "shapeless-compilation"]], "Shapeless Exports": [[519, "shapeless-exports"]], "Simple Example": [[1, "simple-example"]], "Specifying the Stream": [[528, "specifying-the-stream"]], "Supported Data Types": [[347, "id2"]], "TensorFlow": [[524, "tensorflow"]], "The Module Class": [[356, "the-module-class"]], "Thunderbolt Ring": [[518, "thunderbolt-ring"]], "Training Example": [[518, "training-example"]], "Transformations with Compile": [[517, "transformations-with-compile"]], "Transformations with Imported Functions": [[519, "transformations-with-imported-functions"]], "Transforming Compute Graphs": [[523, "transforming-compute-graphs"]], "Transforms": [[515, null]], "Tree Utils": [[516, null]], "Troubleshooting": [[9, "troubleshooting"], [9, "id4"]], "Tuning MPI All Reduce": [[518, "tuning-mpi-all-reduce"]], "Unified Memory": [[527, null]], "Updating the Parameters": [[356, "updating-the-parameters"]], "Usage": [[2, "usage"], [8, null], [522, "usage"]], "Using MLX in C++": [[4, null]], "Using Shape/Strides": [[1, "using-shape-strides"]], "Using Streams": [[528, null]], "Using the Primitive": [[2, "using-the-primitive"]], "Utilizing nn.average_gradients": [[518, "utilizing-nn-average-gradients"]], "Value and Grad": [[356, "value-and-grad"]], "Weight loading and benchmarking": [[6, "weight-loading-and-benchmarking"]], "When to Evaluate": [[523, "when-to-evaluate"]], "Why Lazy Evaluation": [[523, "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": [[344, 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.imag": [[51, null]], "mlx.core.array.item": [[52, null]], "mlx.core.array.itemsize": [[53, null]], "mlx.core.array.log": [[54, null]], "mlx.core.array.log10": [[55, null]], "mlx.core.array.log1p": [[56, null]], "mlx.core.array.log2": [[57, null]], "mlx.core.array.logcumsumexp": [[58, null]], "mlx.core.array.logsumexp": [[59, null]], "mlx.core.array.max": [[60, null]], "mlx.core.array.mean": [[61, null]], "mlx.core.array.min": [[62, null]], "mlx.core.array.moveaxis": [[63, null]], "mlx.core.array.nbytes": [[64, null]], "mlx.core.array.ndim": [[65, null]], "mlx.core.array.prod": [[66, null]], "mlx.core.array.real": [[67, null]], "mlx.core.array.reciprocal": [[68, null]], "mlx.core.array.reshape": [[69, null]], "mlx.core.array.round": [[70, null]], "mlx.core.array.rsqrt": [[71, null]], "mlx.core.array.shape": [[72, null]], "mlx.core.array.sin": [[73, null]], "mlx.core.array.size": [[74, null]], "mlx.core.array.split": [[75, null]], "mlx.core.array.sqrt": [[76, null]], "mlx.core.array.square": [[77, null]], "mlx.core.array.squeeze": [[78, null]], "mlx.core.array.std": [[79, null]], "mlx.core.array.sum": [[80, null]], "mlx.core.array.swapaxes": [[81, null]], "mlx.core.array.tolist": [[82, null]], "mlx.core.array.transpose": [[83, null]], "mlx.core.array.var": [[84, null]], "mlx.core.array.view": [[85, null]], "mlx.core.array_equal": [[86, null]], "mlx.core.as_strided": [[87, null]], "mlx.core.async_eval": [[88, null]], "mlx.core.atleast_1d": [[89, null]], "mlx.core.atleast_2d": [[90, null]], "mlx.core.atleast_3d": [[91, null]], "mlx.core.bitwise_and": [[92, null]], "mlx.core.bitwise_invert": [[93, null]], "mlx.core.bitwise_or": [[94, null]], "mlx.core.bitwise_xor": [[95, null]], "mlx.core.block_masked_mm": [[96, null]], "mlx.core.broadcast_arrays": [[97, null]], "mlx.core.broadcast_to": [[98, null]], "mlx.core.ceil": [[99, null]], "mlx.core.clear_cache": [[100, null]], "mlx.core.clip": [[101, null]], "mlx.core.compile": [[102, null]], "mlx.core.concatenate": [[103, null]], "mlx.core.conj": [[104, null]], "mlx.core.conjugate": [[105, null]], "mlx.core.contiguous": [[106, null]], "mlx.core.conv1d": [[107, null]], "mlx.core.conv2d": [[108, null]], "mlx.core.conv3d": [[109, null]], "mlx.core.conv_general": [[110, null]], "mlx.core.conv_transpose1d": [[111, null]], "mlx.core.conv_transpose2d": [[112, null]], "mlx.core.conv_transpose3d": [[113, null]], "mlx.core.convolve": [[114, null]], "mlx.core.cos": [[115, null]], "mlx.core.cosh": [[116, null]], "mlx.core.cuda.is_available": [[117, null]], "mlx.core.cummax": [[118, null]], "mlx.core.cummin": [[119, null]], "mlx.core.cumprod": [[120, null]], "mlx.core.cumsum": [[121, null]], "mlx.core.custom_function": [[122, null]], "mlx.core.default_device": [[123, null]], "mlx.core.default_stream": [[124, null]], "mlx.core.degrees": [[125, null]], "mlx.core.dequantize": [[126, null]], "mlx.core.diag": [[127, null]], "mlx.core.diagonal": [[128, null]], "mlx.core.disable_compile": [[129, null]], "mlx.core.distributed.Group": [[130, null]], "mlx.core.distributed.all_gather": [[131, null]], "mlx.core.distributed.all_sum": [[132, null]], "mlx.core.distributed.init": [[133, null]], "mlx.core.distributed.is_available": [[134, null]], "mlx.core.distributed.recv": [[135, null]], "mlx.core.distributed.recv_like": [[136, null]], "mlx.core.distributed.send": [[137, null]], "mlx.core.divide": [[138, null]], "mlx.core.divmod": [[139, null]], "mlx.core.einsum": [[140, null]], "mlx.core.einsum_path": [[141, null]], "mlx.core.enable_compile": [[142, null]], "mlx.core.equal": [[143, null]], "mlx.core.erf": [[144, null]], "mlx.core.erfinv": [[145, null]], "mlx.core.eval": [[146, null]], "mlx.core.exp": [[147, null]], "mlx.core.expand_dims": [[148, null]], "mlx.core.expm1": [[149, null]], "mlx.core.export_function": [[150, null]], "mlx.core.export_to_dot": [[151, null]], "mlx.core.exporter": [[152, null]], "mlx.core.eye": [[153, null]], "mlx.core.fast.cuda_kernel": [[154, null]], "mlx.core.fast.layer_norm": [[155, null]], "mlx.core.fast.metal_kernel": [[156, null]], "mlx.core.fast.rms_norm": [[157, null]], "mlx.core.fast.rope": [[158, null]], "mlx.core.fast.scaled_dot_product_attention": [[159, null]], "mlx.core.fft.fft": [[160, null]], "mlx.core.fft.fft2": [[161, null]], "mlx.core.fft.fftn": [[162, null]], "mlx.core.fft.fftshift": [[163, null]], "mlx.core.fft.ifft": [[164, null]], "mlx.core.fft.ifft2": [[165, null]], "mlx.core.fft.ifftn": [[166, null]], "mlx.core.fft.ifftshift": [[167, null]], "mlx.core.fft.irfft": [[168, null]], "mlx.core.fft.irfft2": [[169, null]], "mlx.core.fft.irfftn": [[170, null]], "mlx.core.fft.rfft": [[171, null]], "mlx.core.fft.rfft2": [[172, null]], "mlx.core.fft.rfftn": [[173, null]], "mlx.core.finfo": [[174, null]], "mlx.core.flatten": [[175, null]], "mlx.core.floor": [[176, null]], "mlx.core.floor_divide": [[177, null]], "mlx.core.full": [[178, null]], "mlx.core.gather_mm": [[179, null]], "mlx.core.gather_qmm": [[180, null]], "mlx.core.get_active_memory": [[181, null]], "mlx.core.get_cache_memory": [[182, null]], "mlx.core.get_peak_memory": [[183, null]], "mlx.core.grad": [[184, null]], "mlx.core.greater": [[185, null]], "mlx.core.greater_equal": [[186, null]], "mlx.core.hadamard_transform": [[187, null]], "mlx.core.identity": [[188, null]], "mlx.core.imag": [[189, null]], "mlx.core.import_function": [[190, null]], "mlx.core.inner": [[191, null]], "mlx.core.isclose": [[192, null]], "mlx.core.isfinite": [[193, null]], "mlx.core.isinf": [[194, null]], "mlx.core.isnan": [[195, null]], "mlx.core.isneginf": [[196, null]], "mlx.core.isposinf": [[197, null]], "mlx.core.issubdtype": [[198, null]], "mlx.core.jvp": [[199, null]], "mlx.core.kron": [[200, null]], "mlx.core.left_shift": [[201, null]], "mlx.core.less": [[202, null]], "mlx.core.less_equal": [[203, null]], "mlx.core.linalg.cholesky": [[204, null]], "mlx.core.linalg.cholesky_inv": [[205, null]], "mlx.core.linalg.cross": [[206, null]], "mlx.core.linalg.eig": [[207, null]], "mlx.core.linalg.eigh": [[208, null]], "mlx.core.linalg.eigvals": [[209, null]], "mlx.core.linalg.eigvalsh": [[210, null]], "mlx.core.linalg.inv": [[211, null]], "mlx.core.linalg.lu": [[212, null]], "mlx.core.linalg.lu_factor": [[213, null]], "mlx.core.linalg.norm": [[214, null]], "mlx.core.linalg.pinv": [[215, null]], "mlx.core.linalg.qr": [[216, null]], "mlx.core.linalg.solve": [[217, null]], "mlx.core.linalg.solve_triangular": [[218, null]], "mlx.core.linalg.svd": [[219, null]], "mlx.core.linalg.tri_inv": [[220, null]], "mlx.core.linspace": [[221, null]], "mlx.core.load": [[222, null]], "mlx.core.log": [[223, null]], "mlx.core.log10": [[224, null]], "mlx.core.log1p": [[225, null]], "mlx.core.log2": [[226, null]], "mlx.core.logaddexp": [[227, null]], "mlx.core.logcumsumexp": [[228, null]], "mlx.core.logical_and": [[229, null]], "mlx.core.logical_not": [[230, null]], "mlx.core.logical_or": [[231, null]], "mlx.core.logsumexp": [[232, null]], "mlx.core.matmul": [[233, null]], "mlx.core.max": [[234, null]], "mlx.core.maximum": [[235, null]], "mlx.core.mean": [[236, null]], "mlx.core.meshgrid": [[237, null]], "mlx.core.metal.device_info": [[238, null]], "mlx.core.metal.is_available": [[239, null]], "mlx.core.metal.start_capture": [[240, null]], "mlx.core.metal.stop_capture": [[241, null]], "mlx.core.min": [[242, null]], "mlx.core.minimum": [[243, null]], "mlx.core.moveaxis": [[244, null]], "mlx.core.multiply": [[245, null]], "mlx.core.nan_to_num": [[246, null]], "mlx.core.negative": [[247, null]], "mlx.core.new_stream": [[248, null]], "mlx.core.not_equal": [[249, null]], "mlx.core.ones": [[250, null]], "mlx.core.ones_like": [[251, null]], "mlx.core.outer": [[252, null]], "mlx.core.pad": [[253, null]], "mlx.core.partition": [[254, null]], "mlx.core.power": [[255, null]], "mlx.core.prod": [[256, null]], "mlx.core.put_along_axis": [[257, null]], "mlx.core.quantize": [[258, null]], "mlx.core.quantized_matmul": [[259, null]], "mlx.core.radians": [[260, null]], "mlx.core.random.bernoulli": [[261, null]], "mlx.core.random.categorical": [[262, null]], "mlx.core.random.gumbel": [[263, null]], "mlx.core.random.key": [[264, null]], "mlx.core.random.laplace": [[265, null]], "mlx.core.random.multivariate_normal": [[266, null]], "mlx.core.random.normal": [[267, null]], "mlx.core.random.permutation": [[268, null]], "mlx.core.random.randint": [[269, null]], "mlx.core.random.seed": [[270, null]], "mlx.core.random.split": [[271, null]], "mlx.core.random.truncated_normal": [[272, null]], "mlx.core.random.uniform": [[273, null]], "mlx.core.real": [[274, null]], "mlx.core.reciprocal": [[275, null]], "mlx.core.remainder": [[276, null]], "mlx.core.repeat": [[277, null]], "mlx.core.reset_peak_memory": [[278, null]], "mlx.core.reshape": [[279, null]], "mlx.core.right_shift": [[280, null]], "mlx.core.roll": [[281, null]], "mlx.core.round": [[282, null]], "mlx.core.rsqrt": [[283, null]], "mlx.core.save": [[284, null]], "mlx.core.save_gguf": [[285, null]], "mlx.core.save_safetensors": [[286, null]], "mlx.core.savez": [[287, null]], "mlx.core.savez_compressed": [[288, null]], "mlx.core.set_cache_limit": [[289, null]], "mlx.core.set_default_device": [[290, null]], "mlx.core.set_default_stream": [[291, null]], "mlx.core.set_memory_limit": [[292, null]], "mlx.core.set_wired_limit": [[293, null]], "mlx.core.sigmoid": [[294, null]], "mlx.core.sign": [[295, null]], "mlx.core.sin": [[296, null]], "mlx.core.sinh": [[297, null]], "mlx.core.slice": [[298, null]], "mlx.core.slice_update": [[299, null]], "mlx.core.softmax": [[300, null]], "mlx.core.sort": [[301, null]], "mlx.core.split": [[302, null]], "mlx.core.sqrt": [[303, null]], "mlx.core.square": [[304, null]], "mlx.core.squeeze": [[305, null]], "mlx.core.stack": [[306, null]], "mlx.core.std": [[307, null]], "mlx.core.stop_gradient": [[308, null]], "mlx.core.stream": [[309, null]], "mlx.core.subtract": [[310, null]], "mlx.core.sum": [[311, null]], "mlx.core.swapaxes": [[312, null]], "mlx.core.synchronize": [[313, null]], "mlx.core.take": [[314, null]], "mlx.core.take_along_axis": [[315, null]], "mlx.core.tan": [[316, null]], "mlx.core.tanh": [[317, null]], "mlx.core.tensordot": [[318, null]], "mlx.core.tile": [[319, null]], "mlx.core.topk": [[320, null]], "mlx.core.trace": [[321, null]], "mlx.core.transpose": [[322, null]], "mlx.core.tri": [[323, null]], "mlx.core.tril": [[324, null]], "mlx.core.triu": [[325, null]], "mlx.core.unflatten": [[326, null]], "mlx.core.value_and_grad": [[327, null]], "mlx.core.var": [[328, null]], "mlx.core.view": [[329, null]], "mlx.core.vjp": [[330, null]], "mlx.core.vmap": [[331, null]], "mlx.core.where": [[332, null]], "mlx.core.zeros": [[333, null]], "mlx.core.zeros_like": [[334, null]], "mlx.nn.ALiBi": [[357, null]], "mlx.nn.AvgPool1d": [[358, null]], "mlx.nn.AvgPool2d": [[359, null]], "mlx.nn.AvgPool3d": [[360, null]], "mlx.nn.BatchNorm": [[361, null]], "mlx.nn.CELU": [[362, null]], "mlx.nn.Conv1d": [[363, null]], "mlx.nn.Conv2d": [[364, null]], "mlx.nn.Conv3d": [[365, null]], "mlx.nn.ConvTranspose1d": [[366, null]], "mlx.nn.ConvTranspose2d": [[367, null]], "mlx.nn.ConvTranspose3d": [[368, null]], "mlx.nn.Dropout": [[369, null]], "mlx.nn.Dropout2d": [[370, null]], "mlx.nn.Dropout3d": [[371, null]], "mlx.nn.ELU": [[372, null]], "mlx.nn.Embedding": [[373, null]], "mlx.nn.GELU": [[374, null]], "mlx.nn.GLU": [[375, null]], "mlx.nn.GRU": [[376, null]], "mlx.nn.GroupNorm": [[377, null]], "mlx.nn.HardShrink": [[378, null]], "mlx.nn.HardTanh": [[379, null]], "mlx.nn.Hardswish": [[380, null]], "mlx.nn.InstanceNorm": [[381, null]], "mlx.nn.LSTM": [[382, null]], "mlx.nn.LayerNorm": [[383, null]], "mlx.nn.LeakyReLU": [[384, null]], "mlx.nn.Linear": [[385, null]], "mlx.nn.LogSigmoid": [[386, null]], "mlx.nn.LogSoftmax": [[387, null]], "mlx.nn.MaxPool1d": [[388, null]], "mlx.nn.MaxPool2d": [[389, null]], "mlx.nn.MaxPool3d": [[390, null]], "mlx.nn.Mish": [[391, null]], "mlx.nn.Module.apply": [[392, null]], "mlx.nn.Module.apply_to_modules": [[393, null]], "mlx.nn.Module.children": [[394, null]], "mlx.nn.Module.eval": [[395, null]], "mlx.nn.Module.filter_and_map": [[396, null]], "mlx.nn.Module.freeze": [[397, null]], "mlx.nn.Module.leaf_modules": [[398, null]], "mlx.nn.Module.load_weights": [[399, null]], "mlx.nn.Module.modules": [[400, null]], "mlx.nn.Module.named_modules": [[401, null]], "mlx.nn.Module.parameters": [[402, null]], "mlx.nn.Module.save_weights": [[403, null]], "mlx.nn.Module.set_dtype": [[404, null]], "mlx.nn.Module.state": [[405, null]], "mlx.nn.Module.train": [[406, null]], "mlx.nn.Module.trainable_parameters": [[407, null]], "mlx.nn.Module.training": [[408, null]], "mlx.nn.Module.unfreeze": [[409, null]], "mlx.nn.Module.update": [[410, null]], "mlx.nn.Module.update_modules": [[411, null]], "mlx.nn.MultiHeadAttention": [[412, null]], "mlx.nn.PReLU": [[413, null]], "mlx.nn.QuantizedEmbedding": [[414, null]], "mlx.nn.QuantizedLinear": [[415, null]], "mlx.nn.RMSNorm": [[416, null]], "mlx.nn.RNN": [[417, null]], "mlx.nn.ReLU": [[418, null]], "mlx.nn.ReLU2": [[419, null]], "mlx.nn.ReLU6": [[420, null]], "mlx.nn.RoPE": [[421, null]], "mlx.nn.SELU": [[422, null]], "mlx.nn.Sequential": [[423, null]], "mlx.nn.SiLU": [[424, null]], "mlx.nn.Sigmoid": [[425, null]], "mlx.nn.SinusoidalPositionalEncoding": [[426, null]], "mlx.nn.Softmax": [[427, null]], "mlx.nn.Softmin": [[428, null]], "mlx.nn.Softplus": [[429, null]], "mlx.nn.Softshrink": [[430, null]], "mlx.nn.Softsign": [[431, null]], "mlx.nn.Step": [[432, null]], "mlx.nn.Tanh": [[433, null]], "mlx.nn.Transformer": [[434, null]], "mlx.nn.Upsample": [[435, null]], "mlx.nn.average_gradients": [[335, null]], "mlx.nn.celu": [[444, null]], "mlx.nn.elu": [[445, null]], "mlx.nn.gelu": [[446, null]], "mlx.nn.gelu_approx": [[447, null]], "mlx.nn.gelu_fast_approx": [[448, null]], "mlx.nn.glu": [[449, null]], "mlx.nn.hard_shrink": [[450, null]], "mlx.nn.hard_tanh": [[451, null]], "mlx.nn.hardswish": [[452, null]], "mlx.nn.init.constant": [[436, null]], "mlx.nn.init.glorot_normal": [[437, null]], "mlx.nn.init.glorot_uniform": [[438, null]], "mlx.nn.init.he_normal": [[439, null]], "mlx.nn.init.he_uniform": [[440, null]], "mlx.nn.init.identity": [[441, null]], "mlx.nn.init.normal": [[442, null]], "mlx.nn.init.uniform": [[443, null]], "mlx.nn.leaky_relu": [[453, null]], "mlx.nn.log_sigmoid": [[454, null]], "mlx.nn.log_softmax": [[455, null]], "mlx.nn.losses.binary_cross_entropy": [[456, null]], "mlx.nn.losses.cosine_similarity_loss": [[457, null]], "mlx.nn.losses.cross_entropy": [[458, null]], "mlx.nn.losses.gaussian_nll_loss": [[459, null]], "mlx.nn.losses.hinge_loss": [[460, null]], "mlx.nn.losses.huber_loss": [[461, null]], "mlx.nn.losses.kl_div_loss": [[462, null]], "mlx.nn.losses.l1_loss": [[463, null]], "mlx.nn.losses.log_cosh_loss": [[464, null]], "mlx.nn.losses.margin_ranking_loss": [[465, null]], "mlx.nn.losses.mse_loss": [[466, null]], "mlx.nn.losses.nll_loss": [[467, null]], "mlx.nn.losses.smooth_l1_loss": [[468, null]], "mlx.nn.losses.triplet_loss": [[469, null]], "mlx.nn.mish": [[470, null]], "mlx.nn.prelu": [[471, null]], "mlx.nn.quantize": [[336, null]], "mlx.nn.relu": [[472, null]], "mlx.nn.relu2": [[473, null]], "mlx.nn.relu6": [[474, null]], "mlx.nn.selu": [[475, null]], "mlx.nn.sigmoid": [[476, null]], "mlx.nn.silu": [[477, null]], "mlx.nn.softmax": [[478, null]], "mlx.nn.softmin": [[479, null]], "mlx.nn.softplus": [[480, null]], "mlx.nn.softshrink": [[481, null]], "mlx.nn.step": [[482, null]], "mlx.nn.tanh": [[483, null]], "mlx.nn.value_and_grad": [[337, null]], "mlx.optimizers.AdaDelta": [[491, null]], "mlx.optimizers.Adafactor": [[492, null]], "mlx.optimizers.Adagrad": [[493, null]], "mlx.optimizers.Adam": [[494, null]], "mlx.optimizers.AdamW": [[495, null]], "mlx.optimizers.Adamax": [[496, null]], "mlx.optimizers.Lion": [[497, null]], "mlx.optimizers.MultiOptimizer": [[498, null]], "mlx.optimizers.Muon": [[499, null]], "mlx.optimizers.Optimizer.apply_gradients": [[500, null]], "mlx.optimizers.Optimizer.init": [[501, null]], "mlx.optimizers.Optimizer.state": [[502, null]], "mlx.optimizers.Optimizer.update": [[503, null]], "mlx.optimizers.RMSprop": [[504, null]], "mlx.optimizers.SGD": [[505, null]], "mlx.optimizers.clip_grad_norm": [[338, null]], "mlx.optimizers.cosine_decay": [[506, null]], "mlx.optimizers.exponential_decay": [[507, null]], "mlx.optimizers.join_schedules": [[508, null]], "mlx.optimizers.linear_schedule": [[509, null]], "mlx.optimizers.step_decay": [[510, null]], "mlx.utils.tree_flatten": [[339, null]], "mlx.utils.tree_map": [[340, null]], "mlx.utils.tree_map_with_path": [[341, null]], "mlx.utils.tree_reduce": [[342, null]], "mlx.utils.tree_unflatten": [[343, 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.imag", "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.logcumsumexp", "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.real", "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.cuda.is_available", "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.cuda_kernel", "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.fftshift", "python/_autosummary/mlx.core.fft.ifft", "python/_autosummary/mlx.core.fft.ifft2", "python/_autosummary/mlx.core.fft.ifftn", "python/_autosummary/mlx.core.fft.ifftshift", "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.eig", "python/_autosummary/mlx.core.linalg.eigh", "python/_autosummary/mlx.core.linalg.eigvals", "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.logcumsumexp", "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/cuda", "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.ReLU2", "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.relu2", "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.MultiOptimizer", "python/optimizers/_autosummary/mlx.optimizers.Muon", "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.imag.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.logcumsumexp.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.real.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.cuda.is_available.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.cuda_kernel.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.fftshift.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.ifftshift.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.eig.rst", "python/_autosummary/mlx.core.linalg.eigh.rst", "python/_autosummary/mlx.core.linalg.eigvals.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.logcumsumexp.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/cuda.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.ReLU2.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.relu2.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.MultiOptimizer.rst", "python/optimizers/_autosummary/mlx.optimizers.Muon.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)": [[122, "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)": [[174, "mlx.core.finfo.__init__", false]], "__init__() (group method)": [[130, "mlx.core.distributed.Group.__init__", false]], "__init__() (stream method)": [[344, "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)": [[491, "mlx.optimizers.AdaDelta", false]], "adafactor (class in mlx.optimizers)": [[492, "mlx.optimizers.Adafactor", false]], "adagrad (class in mlx.optimizers)": [[493, "mlx.optimizers.Adagrad", false]], "adam (class in mlx.optimizers)": [[494, "mlx.optimizers.Adam", false]], "adamax (class in mlx.optimizers)": [[496, "mlx.optimizers.Adamax", false]], "adamw (class in mlx.optimizers)": [[495, "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)": [[357, "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)": [[131, "mlx.core.distributed.all_gather", false]], "all_sum() (in module mlx.core.distributed)": [[132, "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)": [[392, "mlx.nn.Module.apply", false]], "apply_gradients() (optimizer method)": [[500, "mlx.optimizers.Optimizer.apply_gradients", false]], "apply_to_modules() (module method)": [[393, "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)": [[86, "mlx.core.array_equal", false]], "as_strided (c++ function)": [[0, "_CPPv410as_strided5array5Shape7Strides6size_t14StreamOrDevice", false]], "as_strided() (in module mlx.core)": [[87, "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)": [[88, "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)": [[89, "mlx.core.atleast_1d", false]], "atleast_2d (c++ function)": [[0, "_CPPv410atleast_2dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_2dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_2d() (in module mlx.core)": [[90, "mlx.core.atleast_2d", false]], "atleast_3d (c++ function)": [[0, "_CPPv410atleast_3dRK5array14StreamOrDevice", false], [0, "_CPPv410atleast_3dRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "atleast_3d() (in module mlx.core)": [[91, "mlx.core.atleast_3d", false]], "average_gradients() (in module mlx.nn)": [[335, "mlx.nn.average_gradients", false]], "avgpool1d (class in mlx.nn)": [[358, "mlx.nn.AvgPool1d", false]], "avgpool2d (class in mlx.nn)": [[359, "mlx.nn.AvgPool2d", false]], "avgpool3d (class in mlx.nn)": [[360, "mlx.nn.AvgPool3d", false]], "batchnorm (class in mlx.nn)": [[361, "mlx.nn.BatchNorm", false]], "bernoulli() (in module mlx.core.random)": [[261, "mlx.core.random.bernoulli", false]], "binary_cross_entropy (class in mlx.nn.losses)": [[456, "mlx.nn.losses.binary_cross_entropy", false]], "bitwise_and (c++ function)": [[0, "_CPPv411bitwise_andRK5arrayRK5array14StreamOrDevice", false]], "bitwise_and() (in module mlx.core)": [[92, "mlx.core.bitwise_and", false]], "bitwise_invert (c++ function)": [[0, "_CPPv414bitwise_invertRK5array14StreamOrDevice", false]], "bitwise_invert() (in module mlx.core)": [[93, "mlx.core.bitwise_invert", false]], "bitwise_or (c++ function)": [[0, "_CPPv410bitwise_orRK5arrayRK5array14StreamOrDevice", false]], "bitwise_or() (in module mlx.core)": [[94, "mlx.core.bitwise_or", false]], "bitwise_xor (c++ function)": [[0, "_CPPv411bitwise_xorRK5arrayRK5array14StreamOrDevice", false]], "bitwise_xor() (in module mlx.core)": [[95, "mlx.core.bitwise_xor", false]], "block_masked_mm (c++ function)": [[0, "_CPPv415block_masked_mm5array5arrayiNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEE14StreamOrDevice", false]], "block_masked_mm() (in module mlx.core)": [[96, "mlx.core.block_masked_mm", false]], "broadcast_arrays (c++ function)": [[0, "_CPPv416broadcast_arraysRKNSt6vectorI5arrayEE14StreamOrDevice", false]], "broadcast_arrays() (in module mlx.core)": [[97, "mlx.core.broadcast_arrays", false]], "broadcast_to (c++ function)": [[0, "_CPPv412broadcast_toRK5arrayRK5Shape14StreamOrDevice", false]], "broadcast_to() (in module mlx.core)": [[98, "mlx.core.broadcast_to", false]], "categorical() (in module mlx.core.random)": [[262, "mlx.core.random.categorical", false]], "ceil (c++ function)": [[0, "_CPPv44ceilRK5array14StreamOrDevice", false]], "ceil() (in module mlx.core)": [[99, "mlx.core.ceil", false]], "celu (class in mlx.nn)": [[362, "mlx.nn.CELU", false], [444, "mlx.nn.celu", false]], "children() (module method)": [[394, "mlx.nn.Module.children", false]], "cholesky() (in module mlx.core.linalg)": [[204, "mlx.core.linalg.cholesky", false]], "cholesky_inv() (in module mlx.core.linalg)": [[205, "mlx.core.linalg.cholesky_inv", false]], "clear_cache() (in module mlx.core)": [[100, "mlx.core.clear_cache", false]], "clip (c++ function)": [[0, "_CPPv44clipRK5arrayRKNSt8optionalI5arrayEERKNSt8optionalI5arrayEE14StreamOrDevice", false]], "clip() (in module mlx.core)": [[101, "mlx.core.clip", false]], "clip_grad_norm() (in module mlx.optimizers)": [[338, "mlx.optimizers.clip_grad_norm", false]], "compile() (in module mlx.core)": [[102, "mlx.core.compile", false]], "concatenate (c++ function)": [[0, "_CPPv411concatenateNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv411concatenateNSt6vectorI5arrayEEi14StreamOrDevice", false]], "concatenate() (in module mlx.core)": [[103, "mlx.core.concatenate", false]], "conj() (array method)": [[40, "mlx.core.array.conj", false]], "conj() (in module mlx.core)": [[104, "mlx.core.conj", false]], "conjugate (c++ function)": [[0, "_CPPv49conjugateRK5array14StreamOrDevice", false]], "conjugate() (in module mlx.core)": [[105, "mlx.core.conjugate", false]], "constant() (in module mlx.nn.init)": [[436, "mlx.nn.init.constant", false]], "contiguous (c++ function)": [[0, "_CPPv410contiguousRK5arrayb14StreamOrDevice", false]], "contiguous() (in module mlx.core)": [[106, "mlx.core.contiguous", false]], "conv1d (c++ function)": [[0, "_CPPv46conv1dRK5arrayRK5arrayiiii14StreamOrDevice", false]], "conv1d (class in mlx.nn)": [[363, "mlx.nn.Conv1d", false]], "conv1d() (in module mlx.core)": [[107, "mlx.core.conv1d", false]], "conv2d (c++ function)": [[0, "_CPPv46conv2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv2d (class in mlx.nn)": [[364, "mlx.nn.Conv2d", false]], "conv2d() (in module mlx.core)": [[108, "mlx.core.conv2d", false]], "conv3d (c++ function)": [[0, "_CPPv46conv3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv3d (class in mlx.nn)": [[365, "mlx.nn.Conv3d", false]], "conv3d() (in module mlx.core)": [[109, "mlx.core.conv3d", false]], "conv_general (c++ function)": [[0, "_CPPv412conv_general5array5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false], [0, "_CPPv412conv_generalRK5arrayRK5arrayNSt6vectorIiEENSt6vectorIiEENSt6vectorIiEENSt6vectorIiEEib14StreamOrDevice", false]], "conv_general() (in module mlx.core)": [[110, "mlx.core.conv_general", false]], "conv_transpose1d (c++ function)": [[0, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", false]], "conv_transpose1d() (in module mlx.core)": [[111, "mlx.core.conv_transpose1d", false]], "conv_transpose2d (c++ function)": [[0, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", false]], "conv_transpose2d() (in module mlx.core)": [[112, "mlx.core.conv_transpose2d", false]], "conv_transpose3d (c++ function)": [[0, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", false]], "conv_transpose3d() (in module mlx.core)": [[113, "mlx.core.conv_transpose3d", false]], "convolve() (in module mlx.core)": [[114, "mlx.core.convolve", false]], "convtranspose1d (class in mlx.nn)": [[366, "mlx.nn.ConvTranspose1d", false]], "convtranspose2d (class in mlx.nn)": [[367, "mlx.nn.ConvTranspose2d", false]], "convtranspose3d (class in mlx.nn)": [[368, "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)": [[115, "mlx.core.cos", false]], "cosh (c++ function)": [[0, "_CPPv44coshRK5array14StreamOrDevice", false]], "cosh() (in module mlx.core)": [[116, "mlx.core.cosh", false]], "cosine_decay() (in module mlx.optimizers)": [[506, "mlx.optimizers.cosine_decay", false]], "cosine_similarity_loss (class in mlx.nn.losses)": [[457, "mlx.nn.losses.cosine_similarity_loss", false]], "cross() (in module mlx.core.linalg)": [[206, "mlx.core.linalg.cross", false]], "cross_entropy (class in mlx.nn.losses)": [[458, "mlx.nn.losses.cross_entropy", false]], "cuda_kernel() (in module mlx.core.fast)": [[154, "mlx.core.fast.cuda_kernel", false]], "cummax (c++ function)": [[0, "_CPPv46cummaxRK5arraybb14StreamOrDevice", false], [0, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", false]], "cummax() (array method)": [[42, "mlx.core.array.cummax", false]], "cummax() (in module mlx.core)": [[118, "mlx.core.cummax", false]], "cummin (c++ function)": [[0, "_CPPv46cumminRK5arraybb14StreamOrDevice", false], [0, "_CPPv46cumminRK5arrayibb14StreamOrDevice", false]], "cummin() (array method)": [[43, "mlx.core.array.cummin", false]], "cummin() (in module mlx.core)": [[119, "mlx.core.cummin", false]], "cumprod (c++ function)": [[0, "_CPPv47cumprodRK5arraybb14StreamOrDevice", false], [0, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", false]], "cumprod() (array method)": [[44, "mlx.core.array.cumprod", false]], "cumprod() (in module mlx.core)": [[120, "mlx.core.cumprod", false]], "cumsum (c++ function)": [[0, "_CPPv46cumsumRK5arraybb14StreamOrDevice", false], [0, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", false]], "cumsum() (array method)": [[45, "mlx.core.array.cumsum", false]], "cumsum() (in module mlx.core)": [[121, "mlx.core.cumsum", false]], "custom_function (class in mlx.core)": [[122, "mlx.core.custom_function", false]], "default_device() (in module mlx.core)": [[123, "mlx.core.default_device", false]], "default_stream() (in module mlx.core)": [[124, "mlx.core.default_stream", false]], "degrees (c++ function)": [[0, "_CPPv47degreesRK5array14StreamOrDevice", false]], "degrees() (in module mlx.core)": [[125, "mlx.core.degrees", false]], "depends (c++ function)": [[0, "_CPPv47dependsRKNSt6vectorI5arrayEERKNSt6vectorI5arrayEE", false]], "dequantize (c++ function)": [[0, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", false]], "dequantize() (in module mlx.core)": [[126, "mlx.core.dequantize", false]], "device (class in mlx.core)": [[10, "mlx.core.Device", false]], "device_info() (in module mlx.core.metal)": [[238, "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)": [[127, "mlx.core.diag", false]], "diagonal (c++ function)": [[0, "_CPPv48diagonalRK5arrayiii14StreamOrDevice", false]], "diagonal() (array method)": [[47, "mlx.core.array.diagonal", false]], "diagonal() (in module mlx.core)": [[128, "mlx.core.diagonal", false]], "disable_compile() (in module mlx.core)": [[129, "mlx.core.disable_compile", false]], "divide (c++ function)": [[0, "_CPPv46divideRK5arrayRK5array14StreamOrDevice", false]], "divide() (in module mlx.core)": [[138, "mlx.core.divide", false]], "divmod (c++ function)": [[0, "_CPPv46divmodRK5arrayRK5array14StreamOrDevice", false]], "divmod() (in module mlx.core)": [[139, "mlx.core.divmod", false]], "dropout (class in mlx.nn)": [[369, "mlx.nn.Dropout", false]], "dropout2d (class in mlx.nn)": [[370, "mlx.nn.Dropout2d", false]], "dropout3d (class in mlx.nn)": [[371, "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]], "eig() (in module mlx.core.linalg)": [[207, "mlx.core.linalg.eig", false]], "eigh() (in module mlx.core.linalg)": [[208, "mlx.core.linalg.eigh", false]], "eigvals() (in module mlx.core.linalg)": [[209, "mlx.core.linalg.eigvals", false]], "eigvalsh() (in module mlx.core.linalg)": [[210, "mlx.core.linalg.eigvalsh", false]], "einsum() (in module mlx.core)": [[140, "mlx.core.einsum", false]], "einsum_path() (in module mlx.core)": [[141, "mlx.core.einsum_path", false]], "elu (class in mlx.nn)": [[372, "mlx.nn.ELU", false], [445, "mlx.nn.elu", false]], "embedding (class in mlx.nn)": [[373, "mlx.nn.Embedding", false]], "enable_compile() (in module mlx.core)": [[142, "mlx.core.enable_compile", false]], "equal (c++ function)": [[0, "_CPPv45equalRK5arrayRK5array14StreamOrDevice", false]], "equal() (in module mlx.core)": [[143, "mlx.core.equal", false]], "erf (c++ function)": [[0, "_CPPv43erfRK5array14StreamOrDevice", false]], "erf() (in module mlx.core)": [[144, "mlx.core.erf", false]], "erfinv (c++ function)": [[0, "_CPPv46erfinvRK5array14StreamOrDevice", false]], "erfinv() (in module mlx.core)": [[145, "mlx.core.erfinv", false]], "eval() (in module mlx.core)": [[146, "mlx.core.eval", false]], "eval() (module method)": [[395, "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)": [[147, "mlx.core.exp", false]], "expand_dims (c++ function)": [[0, "_CPPv411expand_dimsRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv411expand_dimsRK5arrayi14StreamOrDevice", false]], "expand_dims() (in module mlx.core)": [[148, "mlx.core.expand_dims", false]], "expm1 (c++ function)": [[0, "_CPPv45expm1RK5array14StreamOrDevice", false]], "expm1() (in module mlx.core)": [[149, "mlx.core.expm1", false]], "exponential_decay() (in module mlx.optimizers)": [[507, "mlx.optimizers.exponential_decay", false]], "export_function() (in module mlx.core)": [[150, "mlx.core.export_function", false]], "export_to_dot() (in module mlx.core)": [[151, "mlx.core.export_to_dot", false]], "exporter() (in module mlx.core)": [[152, "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)": [[153, "mlx.core.eye", false]], "fft() (in module mlx.core.fft)": [[160, "mlx.core.fft.fft", false]], "fft2() (in module mlx.core.fft)": [[161, "mlx.core.fft.fft2", false]], "fftn() (in module mlx.core.fft)": [[162, "mlx.core.fft.fftn", false]], "fftshift() (in module mlx.core.fft)": [[163, "mlx.core.fft.fftshift", false]], "filter_and_map() (module method)": [[396, "mlx.nn.Module.filter_and_map", false]], "finfo (class in mlx.core)": [[174, "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)": [[175, "mlx.core.flatten", false]], "floor (c++ function)": [[0, "_CPPv45floorRK5array14StreamOrDevice", false]], "floor() (in module mlx.core)": [[176, "mlx.core.floor", false]], "floor_divide (c++ function)": [[0, "_CPPv412floor_divideRK5arrayRK5array14StreamOrDevice", false]], "floor_divide() (in module mlx.core)": [[177, "mlx.core.floor_divide", false]], "freeze() (module method)": [[397, "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)": [[178, "mlx.core.full", false]], "gather (c++ function)": [[0, "_CPPv46gatherRK5arrayRK5arrayiRK5Shape14StreamOrDevice", false], [0, "_CPPv46gatherRK5arrayRKNSt6vectorI5arrayEERKNSt6vectorIiEERK5Shape14StreamOrDevice", false]], "gather_mm (c++ function)": [[0, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", false]], "gather_mm() (in module mlx.core)": [[179, "mlx.core.gather_mm", false]], "gather_qmm (c++ function)": [[0, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", false]], "gather_qmm() (in module mlx.core)": [[180, "mlx.core.gather_qmm", false]], "gaussian_nll_loss (class in mlx.nn.losses)": [[459, "mlx.nn.losses.gaussian_nll_loss", false]], "gelu (class in mlx.nn)": [[374, "mlx.nn.GELU", false], [446, "mlx.nn.gelu", false]], "gelu_approx (class in mlx.nn)": [[447, "mlx.nn.gelu_approx", false]], "gelu_fast_approx (class in mlx.nn)": [[448, "mlx.nn.gelu_fast_approx", false]], "get_active_memory() (in module mlx.core)": [[181, "mlx.core.get_active_memory", false]], "get_cache_memory() (in module mlx.core)": [[182, "mlx.core.get_cache_memory", false]], "get_peak_memory() (in module mlx.core)": [[183, "mlx.core.get_peak_memory", false]], "glorot_normal() (in module mlx.nn.init)": [[437, "mlx.nn.init.glorot_normal", false]], "glorot_uniform() (in module mlx.nn.init)": [[438, "mlx.nn.init.glorot_uniform", false]], "glu (class in mlx.nn)": [[375, "mlx.nn.GLU", false], [449, "mlx.nn.glu", false]], "grad() (in module mlx.core)": [[184, "mlx.core.grad", false]], "greater (c++ function)": [[0, "_CPPv47greaterRK5arrayRK5array14StreamOrDevice", false]], "greater() (in module mlx.core)": [[185, "mlx.core.greater", false]], "greater_equal (c++ function)": [[0, "_CPPv413greater_equalRK5arrayRK5array14StreamOrDevice", false]], "greater_equal() (in module mlx.core)": [[186, "mlx.core.greater_equal", false]], "group (class in mlx.core.distributed)": [[130, "mlx.core.distributed.Group", false]], "groupnorm (class in mlx.nn)": [[377, "mlx.nn.GroupNorm", false]], "gru (class in mlx.nn)": [[376, "mlx.nn.GRU", false]], "gumbel() (in module mlx.core.random)": [[263, "mlx.core.random.gumbel", false]], "hadamard_transform (c++ function)": [[0, "_CPPv418hadamard_transformRK5arrayNSt8optionalIfEE14StreamOrDevice", false]], "hadamard_transform() (in module mlx.core)": [[187, "mlx.core.hadamard_transform", false]], "hard_shrink (class in mlx.nn)": [[450, "mlx.nn.hard_shrink", false]], "hard_tanh (class in mlx.nn)": [[451, "mlx.nn.hard_tanh", false]], "hardshrink (class in mlx.nn)": [[378, "mlx.nn.HardShrink", false]], "hardswish (class in mlx.nn)": [[380, "mlx.nn.Hardswish", false], [452, "mlx.nn.hardswish", false]], "hardtanh (class in mlx.nn)": [[379, "mlx.nn.HardTanh", false]], "he_normal() (in module mlx.nn.init)": [[439, "mlx.nn.init.he_normal", false]], "he_uniform() (in module mlx.nn.init)": [[440, "mlx.nn.init.he_uniform", false]], "hinge_loss (class in mlx.nn.losses)": [[460, "mlx.nn.losses.hinge_loss", false]], "huber_loss (class in mlx.nn.losses)": [[461, "mlx.nn.losses.huber_loss", false]], "identity (c++ function)": [[0, "_CPPv48identityi14StreamOrDevice", false], [0, "_CPPv48identityi5Dtype14StreamOrDevice", false]], "identity() (in module mlx.core)": [[188, "mlx.core.identity", false]], "identity() (in module mlx.nn.init)": [[441, "mlx.nn.init.identity", false]], "ifft() (in module mlx.core.fft)": [[164, "mlx.core.fft.ifft", false]], "ifft2() (in module mlx.core.fft)": [[165, "mlx.core.fft.ifft2", false]], "ifftn() (in module mlx.core.fft)": [[166, "mlx.core.fft.ifftn", false]], "ifftshift() (in module mlx.core.fft)": [[167, "mlx.core.fft.ifftshift", false]], "imag (array property)": [[51, "mlx.core.array.imag", false]], "imag (c++ function)": [[0, "_CPPv44imagRK5array14StreamOrDevice", false]], "imag() (in module mlx.core)": [[189, "mlx.core.imag", false]], "import_function() (in module mlx.core)": [[190, "mlx.core.import_function", false]], "init() (in module mlx.core.distributed)": [[133, "mlx.core.distributed.init", false]], "init() (optimizer method)": [[501, "mlx.optimizers.Optimizer.init", false]], "inner (c++ function)": [[0, "_CPPv45innerRK5arrayRK5array14StreamOrDevice", false]], "inner() (in module mlx.core)": [[191, "mlx.core.inner", false]], "instancenorm (class in mlx.nn)": [[381, "mlx.nn.InstanceNorm", false]], "inv() (in module mlx.core.linalg)": [[211, "mlx.core.linalg.inv", false]], "irfft() (in module mlx.core.fft)": [[168, "mlx.core.fft.irfft", false]], "irfft2() (in module mlx.core.fft)": [[169, "mlx.core.fft.irfft2", false]], "irfftn() (in module mlx.core.fft)": [[170, "mlx.core.fft.irfftn", false]], "is_available() (in module mlx.core.cuda)": [[117, "mlx.core.cuda.is_available", false]], "is_available() (in module mlx.core.distributed)": [[134, "mlx.core.distributed.is_available", false]], "is_available() (in module mlx.core.metal)": [[239, "mlx.core.metal.is_available", false]], "isclose (c++ function)": [[0, "_CPPv47iscloseRK5arrayRK5arrayddb14StreamOrDevice", false]], "isclose() (in module mlx.core)": [[192, "mlx.core.isclose", false]], "isfinite (c++ function)": [[0, "_CPPv48isfiniteRK5array14StreamOrDevice", false]], "isfinite() (in module mlx.core)": [[193, "mlx.core.isfinite", false]], "isinf (c++ function)": [[0, "_CPPv45isinfRK5array14StreamOrDevice", false]], "isinf() (in module mlx.core)": [[194, "mlx.core.isinf", false]], "isnan (c++ function)": [[0, "_CPPv45isnanRK5array14StreamOrDevice", false]], "isnan() (in module mlx.core)": [[195, "mlx.core.isnan", false]], "isneginf (c++ function)": [[0, "_CPPv48isneginfRK5array14StreamOrDevice", false]], "isneginf() (in module mlx.core)": [[196, "mlx.core.isneginf", false]], "isposinf (c++ function)": [[0, "_CPPv48isposinfRK5array14StreamOrDevice", false]], "isposinf() (in module mlx.core)": [[197, "mlx.core.isposinf", false]], "issubdtype() (in module mlx.core)": [[198, "mlx.core.issubdtype", false]], "item() (array method)": [[52, "mlx.core.array.item", false]], "itemsize (array property)": [[53, "mlx.core.array.itemsize", false]], "join_schedules() (in module mlx.optimizers)": [[508, "mlx.optimizers.join_schedules", false]], "jvp() (in module mlx.core)": [[199, "mlx.core.jvp", false]], "key() (in module mlx.core.random)": [[264, "mlx.core.random.key", false]], "kl_div_loss (class in mlx.nn.losses)": [[462, "mlx.nn.losses.kl_div_loss", false]], "kron (c++ function)": [[0, "_CPPv44kronRK5arrayRK5array14StreamOrDevice", false]], "kron() (in module mlx.core)": [[200, "mlx.core.kron", false]], "l1_loss (class in mlx.nn.losses)": [[463, "mlx.nn.losses.l1_loss", false]], "laplace() (in module mlx.core.random)": [[265, "mlx.core.random.laplace", false]], "layer_norm() (in module mlx.core.fast)": [[155, "mlx.core.fast.layer_norm", false]], "layernorm (class in mlx.nn)": [[383, "mlx.nn.LayerNorm", false]], "leaf_modules() (module method)": [[398, "mlx.nn.Module.leaf_modules", false]], "leaky_relu (class in mlx.nn)": [[453, "mlx.nn.leaky_relu", false]], "leakyrelu (class in mlx.nn)": [[384, "mlx.nn.LeakyReLU", false]], "left_shift (c++ function)": [[0, "_CPPv410left_shiftRK5arrayRK5array14StreamOrDevice", false]], "left_shift() (in module mlx.core)": [[201, "mlx.core.left_shift", false]], "less (c++ function)": [[0, "_CPPv44lessRK5arrayRK5array14StreamOrDevice", false]], "less() (in module mlx.core)": [[202, "mlx.core.less", false]], "less_equal (c++ function)": [[0, "_CPPv410less_equalRK5arrayRK5array14StreamOrDevice", false]], "less_equal() (in module mlx.core)": [[203, "mlx.core.less_equal", false]], "linear (class in mlx.nn)": [[385, "mlx.nn.Linear", false]], "linear_schedule() (in module mlx.optimizers)": [[509, "mlx.optimizers.linear_schedule", false]], "linspace (c++ function)": [[0, "_CPPv48linspaceddi5Dtype14StreamOrDevice", false]], "linspace() (in module mlx.core)": [[221, "mlx.core.linspace", false]], "lion (class in mlx.optimizers)": [[497, "mlx.optimizers.Lion", false]], "load() (in module mlx.core)": [[222, "mlx.core.load", false]], "load_weights() (module method)": [[399, "mlx.nn.Module.load_weights", false]], "log (c++ function)": [[0, "_CPPv43logRK5array14StreamOrDevice", false]], "log() (array method)": [[54, "mlx.core.array.log", false]], "log() (in module mlx.core)": [[223, "mlx.core.log", false]], "log10 (c++ function)": [[0, "_CPPv45log10RK5array14StreamOrDevice", false]], "log10() (array method)": [[55, "mlx.core.array.log10", false]], "log10() (in module mlx.core)": [[224, "mlx.core.log10", false]], "log1p (c++ function)": [[0, "_CPPv45log1pRK5array14StreamOrDevice", false]], "log1p() (array method)": [[56, "mlx.core.array.log1p", false]], "log1p() (in module mlx.core)": [[225, "mlx.core.log1p", false]], "log2 (c++ function)": [[0, "_CPPv44log2RK5array14StreamOrDevice", false]], "log2() (array method)": [[57, "mlx.core.array.log2", false]], "log2() (in module mlx.core)": [[226, "mlx.core.log2", false]], "log_cosh_loss (class in mlx.nn.losses)": [[464, "mlx.nn.losses.log_cosh_loss", false]], "log_sigmoid (class in mlx.nn)": [[454, "mlx.nn.log_sigmoid", false]], "log_softmax (class in mlx.nn)": [[455, "mlx.nn.log_softmax", false]], "logaddexp (c++ function)": [[0, "_CPPv49logaddexpRK5arrayRK5array14StreamOrDevice", false]], "logaddexp() (in module mlx.core)": [[227, "mlx.core.logaddexp", false]], "logcumsumexp (c++ function)": [[0, "_CPPv412logcumsumexpRK5arraybb14StreamOrDevice", false], [0, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", false]], "logcumsumexp() (array method)": [[58, "mlx.core.array.logcumsumexp", false]], "logcumsumexp() (in module mlx.core)": [[228, "mlx.core.logcumsumexp", false]], "logical_and (c++ function)": [[0, "_CPPv411logical_andRK5arrayRK5array14StreamOrDevice", false]], "logical_and() (in module mlx.core)": [[229, "mlx.core.logical_and", false]], "logical_not (c++ function)": [[0, "_CPPv411logical_notRK5array14StreamOrDevice", false]], "logical_not() (in module mlx.core)": [[230, "mlx.core.logical_not", false]], "logical_or (c++ function)": [[0, "_CPPv410logical_orRK5arrayRK5array14StreamOrDevice", false]], "logical_or() (in module mlx.core)": [[231, "mlx.core.logical_or", false]], "logsigmoid (class in mlx.nn)": [[386, "mlx.nn.LogSigmoid", false]], "logsoftmax (class in mlx.nn)": [[387, "mlx.nn.LogSoftmax", false]], "logsumexp (c++ function)": [[0, "_CPPv49logsumexpRK5array14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayb14StreamOrDevice", false], [0, "_CPPv49logsumexpRK5arrayib14StreamOrDevice", false]], "logsumexp() (array method)": [[59, "mlx.core.array.logsumexp", false]], "logsumexp() (in module mlx.core)": [[232, "mlx.core.logsumexp", false]], "lstm (class in mlx.nn)": [[382, "mlx.nn.LSTM", false]], "lu() (in module mlx.core.linalg)": [[212, "mlx.core.linalg.lu", false]], "lu_factor() (in module mlx.core.linalg)": [[213, "mlx.core.linalg.lu_factor", false]], "margin_ranking_loss (class in mlx.nn.losses)": [[465, "mlx.nn.losses.margin_ranking_loss", false]], "matmul (c++ function)": [[0, "_CPPv46matmulRK5arrayRK5array14StreamOrDevice", false]], "matmul() (in module mlx.core)": [[233, "mlx.core.matmul", false]], "max (c++ function)": [[0, "_CPPv43maxRK5array14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayb14StreamOrDevice", false], [0, "_CPPv43maxRK5arrayib14StreamOrDevice", false]], "max() (array method)": [[60, "mlx.core.array.max", false]], "max() (in module mlx.core)": [[234, "mlx.core.max", false]], "maximum (c++ function)": [[0, "_CPPv47maximumRK5arrayRK5array14StreamOrDevice", false]], "maximum() (in module mlx.core)": [[235, "mlx.core.maximum", false]], "maxpool1d (class in mlx.nn)": [[388, "mlx.nn.MaxPool1d", false]], "maxpool2d (class in mlx.nn)": [[389, "mlx.nn.MaxPool2d", false]], "maxpool3d (class in mlx.nn)": [[390, "mlx.nn.MaxPool3d", false]], "mean (c++ function)": [[0, "_CPPv44meanRK5array14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayb14StreamOrDevice", false], [0, "_CPPv44meanRK5arrayib14StreamOrDevice", false]], "mean() (array method)": [[61, "mlx.core.array.mean", false]], "mean() (in module mlx.core)": [[236, "mlx.core.mean", false]], "meshgrid (c++ function)": [[0, "_CPPv48meshgridRKNSt6vectorI5arrayEEbRKNSt6stringE14StreamOrDevice", false]], "meshgrid() (in module mlx.core)": [[237, "mlx.core.meshgrid", false]], "metal_kernel() (in module mlx.core.fast)": [[156, "mlx.core.fast.metal_kernel", false]], "min (c++ function)": [[0, "_CPPv43minRK5array14StreamOrDevice", false], [0, "_CPPv43minRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayb14StreamOrDevice", false], [0, "_CPPv43minRK5arrayib14StreamOrDevice", false]], "min() (array method)": [[62, "mlx.core.array.min", false]], "min() (in module mlx.core)": [[242, "mlx.core.min", false]], "minimum (c++ function)": [[0, "_CPPv47minimumRK5arrayRK5array14StreamOrDevice", false]], "minimum() (in module mlx.core)": [[243, "mlx.core.minimum", false]], "mish (class in mlx.nn)": [[391, "mlx.nn.Mish", false], [470, "mlx.nn.mish", false]], "module (class in mlx.nn)": [[488, "mlx.nn.Module", false]], "modules() (module method)": [[400, "mlx.nn.Module.modules", false]], "moveaxis (c++ function)": [[0, "_CPPv48moveaxisRK5arrayii14StreamOrDevice", false]], "moveaxis() (array method)": [[63, "mlx.core.array.moveaxis", false]], "moveaxis() (in module mlx.core)": [[244, "mlx.core.moveaxis", false]], "mse_loss (class in mlx.nn.losses)": [[466, "mlx.nn.losses.mse_loss", false]], "multiheadattention (class in mlx.nn)": [[412, "mlx.nn.MultiHeadAttention", false]], "multioptimizer (class in mlx.optimizers)": [[498, "mlx.optimizers.MultiOptimizer", false]], "multiply (c++ function)": [[0, "_CPPv48multiplyRK5arrayRK5array14StreamOrDevice", false]], "multiply() (in module mlx.core)": [[245, "mlx.core.multiply", false]], "multivariate_normal() (in module mlx.core.random)": [[266, "mlx.core.random.multivariate_normal", false]], "muon (class in mlx.optimizers)": [[499, "mlx.optimizers.Muon", false]], "named_modules() (module method)": [[401, "mlx.nn.Module.named_modules", false]], "nan_to_num (c++ function)": [[0, "_CPPv410nan_to_numRK5arrayfKNSt8optionalIfEEKNSt8optionalIfEE14StreamOrDevice", false]], "nan_to_num() (in module mlx.core)": [[246, "mlx.core.nan_to_num", false]], "nbytes (array property)": [[64, "mlx.core.array.nbytes", false]], "ndim (array property)": [[65, "mlx.core.array.ndim", false]], "negative (c++ function)": [[0, "_CPPv48negativeRK5array14StreamOrDevice", false]], "negative() (in module mlx.core)": [[247, "mlx.core.negative", false]], "new_stream() (in module mlx.core)": [[248, "mlx.core.new_stream", false]], "nll_loss (class in mlx.nn.losses)": [[467, "mlx.nn.losses.nll_loss", false]], "norm() (in module mlx.core.linalg)": [[214, "mlx.core.linalg.norm", false]], "normal() (in module mlx.core.random)": [[267, "mlx.core.random.normal", false]], "normal() (in module mlx.nn.init)": [[442, "mlx.nn.init.normal", false]], "not_equal (c++ function)": [[0, "_CPPv49not_equalRK5arrayRK5array14StreamOrDevice", false]], "not_equal() (in module mlx.core)": [[249, "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)": [[250, "mlx.core.ones", false]], "ones_like (c++ function)": [[0, "_CPPv49ones_likeRK5array14StreamOrDevice", false]], "ones_like() (in module mlx.core)": [[251, "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)": [[512, "mlx.optimizers.Optimizer", false]], "outer (c++ function)": [[0, "_CPPv45outerRK5arrayRK5array14StreamOrDevice", false]], "outer() (in module mlx.core)": [[252, "mlx.core.outer", false]], "pad (c++ function)": [[0, "_CPPv43padRK5arrayRKNSt4pairIiiEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorINSt4pairIiiEEEERK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayRKNSt6vectorIiEERK5ShapeRK5ShapeRK5arrayRKNSt6stringE14StreamOrDevice", false], [0, "_CPPv43padRK5arrayiRK5arrayRKNSt6stringE14StreamOrDevice", false]], "pad() (in module mlx.core)": [[253, "mlx.core.pad", false]], "parameters() (module method)": [[402, "mlx.nn.Module.parameters", false]], "partition (c++ function)": [[0, "_CPPv49partitionRK5arrayi14StreamOrDevice", false], [0, "_CPPv49partitionRK5arrayii14StreamOrDevice", false]], "partition() (in module mlx.core)": [[254, "mlx.core.partition", false]], "permutation() (in module mlx.core.random)": [[268, "mlx.core.random.permutation", false]], "pinv() (in module mlx.core.linalg)": [[215, "mlx.core.linalg.pinv", false]], "power (c++ function)": [[0, "_CPPv45powerRK5arrayRK5array14StreamOrDevice", false]], "power() (in module mlx.core)": [[255, "mlx.core.power", false]], "prelu (class in mlx.nn)": [[413, "mlx.nn.PReLU", false], [471, "mlx.nn.prelu", false]], "prod (c++ function)": [[0, "_CPPv44prodRK5array14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayb14StreamOrDevice", false], [0, "_CPPv44prodRK5arrayib14StreamOrDevice", false]], "prod() (array method)": [[66, "mlx.core.array.prod", false]], "prod() (in module mlx.core)": [[256, "mlx.core.prod", false]], "put_along_axis (c++ function)": [[0, "_CPPv414put_along_axisRK5arrayRK5arrayRK5arrayi14StreamOrDevice", false]], "put_along_axis() (in module mlx.core)": [[257, "mlx.core.put_along_axis", false]], "qr() (in module mlx.core.linalg)": [[216, "mlx.core.linalg.qr", false]], "quantize (c++ function)": [[0, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", false]], "quantize() (in module mlx.core)": [[258, "mlx.core.quantize", false]], "quantize() (in module mlx.nn)": [[336, "mlx.nn.quantize", false]], "quantized_matmul (c++ function)": [[0, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", false]], "quantized_matmul() (in module mlx.core)": [[259, "mlx.core.quantized_matmul", false]], "quantizedembedding (class in mlx.nn)": [[414, "mlx.nn.QuantizedEmbedding", false]], "quantizedlinear (class in mlx.nn)": [[415, "mlx.nn.QuantizedLinear", false]], "radians (c++ function)": [[0, "_CPPv47radiansRK5array14StreamOrDevice", false]], "radians() (in module mlx.core)": [[260, "mlx.core.radians", false]], "randint() (in module mlx.core.random)": [[269, "mlx.core.random.randint", false]], "real (array property)": [[67, "mlx.core.array.real", false]], "real (c++ function)": [[0, "_CPPv44realRK5array14StreamOrDevice", false]], "real() (in module mlx.core)": [[274, "mlx.core.real", false]], "reciprocal (c++ function)": [[0, "_CPPv410reciprocalRK5array14StreamOrDevice", false]], "reciprocal() (array method)": [[68, "mlx.core.array.reciprocal", false]], "reciprocal() (in module mlx.core)": [[275, "mlx.core.reciprocal", false]], "recv() (in module mlx.core.distributed)": [[135, "mlx.core.distributed.recv", false]], "recv_like() (in module mlx.core.distributed)": [[136, "mlx.core.distributed.recv_like", false]], "relu (class in mlx.nn)": [[418, "mlx.nn.ReLU", false], [472, "mlx.nn.relu", false]], "relu2 (class in mlx.nn)": [[419, "mlx.nn.ReLU2", false], [473, "mlx.nn.relu2", false]], "relu6 (class in mlx.nn)": [[420, "mlx.nn.ReLU6", false], [474, "mlx.nn.relu6", false]], "remainder (c++ function)": [[0, "_CPPv49remainderRK5arrayRK5array14StreamOrDevice", false]], "remainder() (in module mlx.core)": [[276, "mlx.core.remainder", false]], "repeat (c++ function)": [[0, "_CPPv46repeatRK5arrayi14StreamOrDevice", false], [0, "_CPPv46repeatRK5arrayii14StreamOrDevice", false]], "repeat() (in module mlx.core)": [[277, "mlx.core.repeat", false]], "reset_peak_memory() (in module mlx.core)": [[278, "mlx.core.reset_peak_memory", false]], "reshape (c++ function)": [[0, "_CPPv47reshapeRK5array5Shape14StreamOrDevice", false]], "reshape() (array method)": [[69, "mlx.core.array.reshape", false]], "reshape() (in module mlx.core)": [[279, "mlx.core.reshape", false]], "rfft() (in module mlx.core.fft)": [[171, "mlx.core.fft.rfft", false]], "rfft2() (in module mlx.core.fft)": [[172, "mlx.core.fft.rfft2", false]], "rfftn() (in module mlx.core.fft)": [[173, "mlx.core.fft.rfftn", false]], "right_shift (c++ function)": [[0, "_CPPv411right_shiftRK5arrayRK5array14StreamOrDevice", false]], "right_shift() (in module mlx.core)": [[280, "mlx.core.right_shift", false]], "rms_norm() (in module mlx.core.fast)": [[157, "mlx.core.fast.rms_norm", false]], "rmsnorm (class in mlx.nn)": [[416, "mlx.nn.RMSNorm", false]], "rmsprop (class in mlx.optimizers)": [[504, "mlx.optimizers.RMSprop", false]], "rnn (class in mlx.nn)": [[417, "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)": [[281, "mlx.core.roll", false]], "rope (class in mlx.nn)": [[421, "mlx.nn.RoPE", false]], "rope() (in module mlx.core.fast)": [[158, "mlx.core.fast.rope", false]], "round (c++ function)": [[0, "_CPPv45roundRK5array14StreamOrDevice", false], [0, "_CPPv45roundRK5arrayi14StreamOrDevice", false]], "round() (array method)": [[70, "mlx.core.array.round", false]], "round() (in module mlx.core)": [[282, "mlx.core.round", false]], "rsqrt (c++ function)": [[0, "_CPPv45rsqrtRK5array14StreamOrDevice", false]], "rsqrt() (array method)": [[71, "mlx.core.array.rsqrt", false]], "rsqrt() (in module mlx.core)": [[283, "mlx.core.rsqrt", false]], "save() (in module mlx.core)": [[284, "mlx.core.save", false]], "save_gguf() (in module mlx.core)": [[285, "mlx.core.save_gguf", false]], "save_safetensors() (in module mlx.core)": [[286, "mlx.core.save_safetensors", false]], "save_weights() (module method)": [[403, "mlx.nn.Module.save_weights", false]], "savez() (in module mlx.core)": [[287, "mlx.core.savez", false]], "savez_compressed() (in module mlx.core)": [[288, "mlx.core.savez_compressed", false]], "scaled_dot_product_attention() (in module mlx.core.fast)": [[159, "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)": [[270, "mlx.core.random.seed", false]], "segmented_mm (c++ function)": [[0, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", false]], "selu (class in mlx.nn)": [[422, "mlx.nn.SELU", false], [475, "mlx.nn.selu", false]], "send() (in module mlx.core.distributed)": [[137, "mlx.core.distributed.send", false]], "sequential (class in mlx.nn)": [[423, "mlx.nn.Sequential", false]], "set_cache_limit() (in module mlx.core)": [[289, "mlx.core.set_cache_limit", false]], "set_default_device() (in module mlx.core)": [[290, "mlx.core.set_default_device", false]], "set_default_stream() (in module mlx.core)": [[291, "mlx.core.set_default_stream", false]], "set_dtype() (module method)": [[404, "mlx.nn.Module.set_dtype", false]], "set_memory_limit() (in module mlx.core)": [[292, "mlx.core.set_memory_limit", false]], "set_wired_limit() (in module mlx.core)": [[293, "mlx.core.set_wired_limit", false]], "sgd (class in mlx.optimizers)": [[505, "mlx.optimizers.SGD", false]], "shape (array property)": [[72, "mlx.core.array.shape", false]], "sigmoid (c++ function)": [[0, "_CPPv47sigmoidRK5array14StreamOrDevice", false]], "sigmoid (class in mlx.nn)": [[425, "mlx.nn.Sigmoid", false], [476, "mlx.nn.sigmoid", false]], "sigmoid() (in module mlx.core)": [[294, "mlx.core.sigmoid", false]], "sign (c++ function)": [[0, "_CPPv44signRK5array14StreamOrDevice", false]], "sign() (in module mlx.core)": [[295, "mlx.core.sign", false]], "silu (class in mlx.nn)": [[424, "mlx.nn.SiLU", false], [477, "mlx.nn.silu", false]], "sin (c++ function)": [[0, "_CPPv43sinRK5array14StreamOrDevice", false]], "sin() (array method)": [[73, "mlx.core.array.sin", false]], "sin() (in module mlx.core)": [[296, "mlx.core.sin", false]], "sinh (c++ function)": [[0, "_CPPv44sinhRK5array14StreamOrDevice", false]], "sinh() (in module mlx.core)": [[297, "mlx.core.sinh", false]], "sinusoidalpositionalencoding (class in mlx.nn)": [[426, "mlx.nn.SinusoidalPositionalEncoding", false]], "size (array property)": [[74, "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)": [[298, "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)": [[299, "mlx.core.slice_update", false]], "smooth_l1_loss (class in mlx.nn.losses)": [[468, "mlx.nn.losses.smooth_l1_loss", false]], "softmax (c++ function)": [[0, "_CPPv47softmaxRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayb14StreamOrDevice", false], [0, "_CPPv47softmaxRK5arrayib14StreamOrDevice", false]], "softmax (class in mlx.nn)": [[427, "mlx.nn.Softmax", false], [478, "mlx.nn.softmax", false]], "softmax() (in module mlx.core)": [[300, "mlx.core.softmax", false]], "softmin (class in mlx.nn)": [[428, "mlx.nn.Softmin", false], [479, "mlx.nn.softmin", false]], "softplus (class in mlx.nn)": [[429, "mlx.nn.Softplus", false], [480, "mlx.nn.softplus", false]], "softshrink (class in mlx.nn)": [[430, "mlx.nn.Softshrink", false], [481, "mlx.nn.softshrink", false]], "softsign (class in mlx.nn)": [[431, "mlx.nn.Softsign", false]], "solve() (in module mlx.core.linalg)": [[217, "mlx.core.linalg.solve", false]], "solve_triangular() (in module mlx.core.linalg)": [[218, "mlx.core.linalg.solve_triangular", false]], "sort (c++ function)": [[0, "_CPPv44sortRK5array14StreamOrDevice", false], [0, "_CPPv44sortRK5arrayi14StreamOrDevice", false]], "sort() (in module mlx.core)": [[301, "mlx.core.sort", false]], "split (c++ function)": [[0, "_CPPv45splitRK5arrayRK5Shape14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayRK5Shapei14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayi14StreamOrDevice", false], [0, "_CPPv45splitRK5arrayii14StreamOrDevice", false]], "split() (array method)": [[75, "mlx.core.array.split", false]], "split() (in module mlx.core)": [[302, "mlx.core.split", false]], "split() (in module mlx.core.random)": [[271, "mlx.core.random.split", false]], "sqrt (c++ function)": [[0, "_CPPv44sqrtRK5array14StreamOrDevice", false]], "sqrt() (array method)": [[76, "mlx.core.array.sqrt", false]], "sqrt() (in module mlx.core)": [[303, "mlx.core.sqrt", false]], "square (c++ function)": [[0, "_CPPv46squareRK5array14StreamOrDevice", false]], "square() (array method)": [[77, "mlx.core.array.square", false]], "square() (in module mlx.core)": [[304, "mlx.core.square", false]], "squeeze (c++ function)": [[0, "_CPPv47squeezeRK5array14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayRKNSt6vectorIiEE14StreamOrDevice", false], [0, "_CPPv47squeezeRK5arrayi14StreamOrDevice", false]], "squeeze() (array method)": [[78, "mlx.core.array.squeeze", false]], "squeeze() (in module mlx.core)": [[305, "mlx.core.squeeze", false]], "stack (c++ function)": [[0, "_CPPv45stackRKNSt6vectorI5arrayEE14StreamOrDevice", false], [0, "_CPPv45stackRKNSt6vectorI5arrayEEi14StreamOrDevice", false]], "stack() (in module mlx.core)": [[306, "mlx.core.stack", false]], "start_capture() (in module mlx.core.metal)": [[240, "mlx.core.metal.start_capture", false]], "state (module property)": [[405, "mlx.nn.Module.state", false]], "state (optimizer property)": [[502, "mlx.optimizers.Optimizer.state", false]], "std (c++ function)": [[0, "_CPPv4StRK5array14StreamOrDevice", false], [0, "_CPPv4StRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv4StRK5arraybi14StreamOrDevice", false], [0, "_CPPv4StRK5arrayibi14StreamOrDevice", false]], "std() (array method)": [[79, "mlx.core.array.std", false]], "std() (in module mlx.core)": [[307, "mlx.core.std", false]], "step (class in mlx.nn)": [[432, "mlx.nn.Step", false], [482, "mlx.nn.step", false]], "step_decay() (in module mlx.optimizers)": [[510, "mlx.optimizers.step_decay", false]], "stop_capture() (in module mlx.core.metal)": [[241, "mlx.core.metal.stop_capture", false]], "stop_gradient (c++ function)": [[0, "_CPPv413stop_gradientRK5array14StreamOrDevice", false]], "stop_gradient() (in module mlx.core)": [[308, "mlx.core.stop_gradient", false]], "stream (class in mlx.core)": [[344, "mlx.core.Stream", false]], "stream() (in module mlx.core)": [[309, "mlx.core.stream", false]], "subtract (c++ function)": [[0, "_CPPv48subtractRK5arrayRK5array14StreamOrDevice", false]], "subtract() (in module mlx.core)": [[310, "mlx.core.subtract", false]], "sum (c++ function)": [[0, "_CPPv43sumRK5array14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayRKNSt6vectorIiEEb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayb14StreamOrDevice", false], [0, "_CPPv43sumRK5arrayib14StreamOrDevice", false]], "sum() (array method)": [[80, "mlx.core.array.sum", false]], "sum() (in module mlx.core)": [[311, "mlx.core.sum", false]], "svd() (in module mlx.core.linalg)": [[219, "mlx.core.linalg.svd", false]], "swapaxes (c++ function)": [[0, "_CPPv48swapaxesRK5arrayii14StreamOrDevice", false]], "swapaxes() (array method)": [[81, "mlx.core.array.swapaxes", false]], "swapaxes() (in module mlx.core)": [[312, "mlx.core.swapaxes", false]], "synchronize() (in module mlx.core)": [[313, "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)": [[314, "mlx.core.take", false]], "take_along_axis (c++ function)": [[0, "_CPPv415take_along_axisRK5arrayRK5arrayi14StreamOrDevice", false]], "take_along_axis() (in module mlx.core)": [[315, "mlx.core.take_along_axis", false]], "tan (c++ function)": [[0, "_CPPv43tanRK5array14StreamOrDevice", false]], "tan() (in module mlx.core)": [[316, "mlx.core.tan", false]], "tanh (c++ function)": [[0, "_CPPv44tanhRK5array14StreamOrDevice", false]], "tanh (class in mlx.nn)": [[433, "mlx.nn.Tanh", false], [483, "mlx.nn.tanh", false]], "tanh() (in module mlx.core)": [[317, "mlx.core.tanh", false]], "tensordot (c++ function)": [[0, "_CPPv49tensordotRK5arrayRK5arrayKi14StreamOrDevice", false], [0, "_CPPv49tensordotRK5arrayRK5arrayRKNSt6vectorIiEERKNSt6vectorIiEE14StreamOrDevice", false]], "tensordot() (in module mlx.core)": [[318, "mlx.core.tensordot", false]], "tile (c++ function)": [[0, "_CPPv44tileRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "tile() (in module mlx.core)": [[319, "mlx.core.tile", false]], "tolist() (array method)": [[82, "mlx.core.array.tolist", false]], "topk (c++ function)": [[0, "_CPPv44topkRK5arrayi14StreamOrDevice", false], [0, "_CPPv44topkRK5arrayii14StreamOrDevice", false]], "topk() (in module mlx.core)": [[320, "mlx.core.topk", false]], "trace (c++ function)": [[0, "_CPPv45traceRK5array14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii14StreamOrDevice", false], [0, "_CPPv45traceRK5arrayiii5Dtype14StreamOrDevice", false]], "trace() (in module mlx.core)": [[321, "mlx.core.trace", false]], "train() (module method)": [[406, "mlx.nn.Module.train", false]], "trainable_parameters() (module method)": [[407, "mlx.nn.Module.trainable_parameters", false]], "training (module property)": [[408, "mlx.nn.Module.training", false]], "transformer (class in mlx.nn)": [[434, "mlx.nn.Transformer", false]], "transpose (c++ function)": [[0, "_CPPv49transposeRK5array14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt16initializer_listIiEE14StreamOrDevice", false], [0, "_CPPv49transposeRK5arrayNSt6vectorIiEE14StreamOrDevice", false]], "transpose() (array method)": [[83, "mlx.core.array.transpose", false]], "transpose() (in module mlx.core)": [[322, "mlx.core.transpose", false]], "tree_flatten() (in module mlx.utils)": [[339, "mlx.utils.tree_flatten", false]], "tree_map() (in module mlx.utils)": [[340, "mlx.utils.tree_map", false]], "tree_map_with_path() (in module mlx.utils)": [[341, "mlx.utils.tree_map_with_path", false]], "tree_reduce() (in module mlx.utils)": [[342, "mlx.utils.tree_reduce", false]], "tree_unflatten() (in module mlx.utils)": [[343, "mlx.utils.tree_unflatten", false]], "tri (c++ function)": [[0, "_CPPv43trii5Dtype14StreamOrDevice", false], [0, "_CPPv43triiii5Dtype14StreamOrDevice", false]], "tri() (in module mlx.core)": [[323, "mlx.core.tri", false]], "tri_inv() (in module mlx.core.linalg)": [[220, "mlx.core.linalg.tri_inv", false]], "tril (c++ function)": [[0, "_CPPv44tril5arrayi14StreamOrDevice", false]], "tril() (in module mlx.core)": [[324, "mlx.core.tril", false]], "triplet_loss (class in mlx.nn.losses)": [[469, "mlx.nn.losses.triplet_loss", false]], "triu (c++ function)": [[0, "_CPPv44triu5arrayi14StreamOrDevice", false]], "triu() (in module mlx.core)": [[325, "mlx.core.triu", false]], "truncated_normal() (in module mlx.core.random)": [[272, "mlx.core.random.truncated_normal", false]], "unflatten (c++ function)": [[0, "_CPPv49unflattenRK5arrayi5Shape14StreamOrDevice", false]], "unflatten() (in module mlx.core)": [[326, "mlx.core.unflatten", false]], "unfreeze() (module method)": [[409, "mlx.nn.Module.unfreeze", false]], "uniform() (in module mlx.core.random)": [[273, "mlx.core.random.uniform", false]], "uniform() (in module mlx.nn.init)": [[443, "mlx.nn.init.uniform", false]], "update() (module method)": [[410, "mlx.nn.Module.update", false]], "update() (optimizer method)": [[503, "mlx.optimizers.Optimizer.update", false]], "update_modules() (module method)": [[411, "mlx.nn.Module.update_modules", false]], "upsample (class in mlx.nn)": [[435, "mlx.nn.Upsample", false]], "value_and_grad() (in module mlx.core)": [[327, "mlx.core.value_and_grad", false]], "value_and_grad() (in module mlx.nn)": [[337, "mlx.nn.value_and_grad", false]], "var (c++ function)": [[0, "_CPPv43varRK5array14StreamOrDevice", false], [0, "_CPPv43varRK5arrayRKNSt6vectorIiEEbi14StreamOrDevice", false], [0, "_CPPv43varRK5arraybi14StreamOrDevice", false], [0, "_CPPv43varRK5arrayibi14StreamOrDevice", false]], "var() (array method)": [[84, "mlx.core.array.var", false]], "var() (in module mlx.core)": [[328, "mlx.core.var", false]], "view (c++ function)": [[0, "_CPPv44viewRK5arrayRK5Dtype14StreamOrDevice", false]], "view() (array method)": [[85, "mlx.core.array.view", false]], "view() (in module mlx.core)": [[329, "mlx.core.view", false]], "vjp() (in module mlx.core)": [[330, "mlx.core.vjp", false]], "vmap() (in module mlx.core)": [[331, "mlx.core.vmap", false]], "where (c++ function)": [[0, "_CPPv45whereRK5arrayRK5arrayRK5array14StreamOrDevice", false]], "where() (in module mlx.core)": [[332, "mlx.core.where", false]], "zeros (c++ function)": [[0, "_CPPv45zerosRK5Shape14StreamOrDevice", false], [0, "_CPPv45zerosRK5Shape5Dtype14StreamOrDevice", false]], "zeros() (in module mlx.core)": [[333, "mlx.core.zeros", false]], "zeros_like (c++ function)": [[0, "_CPPv410zeros_likeRK5array14StreamOrDevice", false]], "zeros_like() (in module mlx.core)": [[334, "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_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::dilation"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::groups"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::input"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::output_padding"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::padding"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::s"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::stride"], [0, 1, 1, "_CPPv416conv_transpose1dRK5arrayRK5arrayiiiii14StreamOrDevice", "conv_transpose1d::weight"], [0, 0, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::dilation"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::groups"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::input"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::output_padding"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::padding"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::s"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::stride"], [0, 1, 1, "_CPPv416conv_transpose2dRK5arrayRK5arrayRKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEERKNSt4pairIiiEEi14StreamOrDevice", "conv_transpose2d::weight"], [0, 0, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::dilation"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::groups"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::input"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::output_padding"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::padding"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::s"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "conv_transpose3d::stride"], [0, 1, 1, "_CPPv416conv_transpose3dRK5arrayRK5arrayRKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEERKNSt5tupleIiiiEEi14StreamOrDevice", "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, "_CPPv46cummaxRK5arraybb14StreamOrDevice", "cummax"], [0, 0, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax"], [0, 1, 1, "_CPPv46cummaxRK5arraybb14StreamOrDevice", "cummax::a"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::a"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::axis"], [0, 1, 1, "_CPPv46cummaxRK5arraybb14StreamOrDevice", "cummax::inclusive"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::inclusive"], [0, 1, 1, "_CPPv46cummaxRK5arraybb14StreamOrDevice", "cummax::reverse"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::reverse"], [0, 1, 1, "_CPPv46cummaxRK5arraybb14StreamOrDevice", "cummax::s"], [0, 1, 1, "_CPPv46cummaxRK5arrayibb14StreamOrDevice", "cummax::s"], [0, 0, 1, "_CPPv46cumminRK5arraybb14StreamOrDevice", "cummin"], [0, 0, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin"], [0, 1, 1, "_CPPv46cumminRK5arraybb14StreamOrDevice", "cummin::a"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::a"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::axis"], [0, 1, 1, "_CPPv46cumminRK5arraybb14StreamOrDevice", "cummin::inclusive"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::inclusive"], [0, 1, 1, "_CPPv46cumminRK5arraybb14StreamOrDevice", "cummin::reverse"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::reverse"], [0, 1, 1, "_CPPv46cumminRK5arraybb14StreamOrDevice", "cummin::s"], [0, 1, 1, "_CPPv46cumminRK5arrayibb14StreamOrDevice", "cummin::s"], [0, 0, 1, "_CPPv47cumprodRK5arraybb14StreamOrDevice", "cumprod"], [0, 0, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod"], [0, 1, 1, "_CPPv47cumprodRK5arraybb14StreamOrDevice", "cumprod::a"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::a"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::axis"], [0, 1, 1, "_CPPv47cumprodRK5arraybb14StreamOrDevice", "cumprod::inclusive"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::inclusive"], [0, 1, 1, "_CPPv47cumprodRK5arraybb14StreamOrDevice", "cumprod::reverse"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::reverse"], [0, 1, 1, "_CPPv47cumprodRK5arraybb14StreamOrDevice", "cumprod::s"], [0, 1, 1, "_CPPv47cumprodRK5arrayibb14StreamOrDevice", "cumprod::s"], [0, 0, 1, "_CPPv46cumsumRK5arraybb14StreamOrDevice", "cumsum"], [0, 0, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum"], [0, 1, 1, "_CPPv46cumsumRK5arraybb14StreamOrDevice", "cumsum::a"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::a"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::axis"], [0, 1, 1, "_CPPv46cumsumRK5arraybb14StreamOrDevice", "cumsum::inclusive"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::inclusive"], [0, 1, 1, "_CPPv46cumsumRK5arraybb14StreamOrDevice", "cumsum::reverse"], [0, 1, 1, "_CPPv46cumsumRK5arrayibb14StreamOrDevice", "cumsum::reverse"], [0, 1, 1, "_CPPv46cumsumRK5arraybb14StreamOrDevice", "cumsum::s"], [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, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::biases"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::bits"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::group_size"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::mode"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::s"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "dequantize::scales"], [0, 1, 1, "_CPPv410dequantizeRK5arrayRK5arrayRKNSt8optionalI5arrayEEiiRKNSt6stringE14StreamOrDevice", "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_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::a"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::b"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::lhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::rhs_indices"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::s"], [0, 1, 1, "_CPPv49gather_mm5array5arrayNSt8optionalI5arrayEENSt8optionalI5arrayEEb14StreamOrDevice", "gather_mm::sorted_indices"], [0, 0, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::biases"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::bits"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::group_size"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::lhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::mode"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::rhs_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::s"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::scales"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::sorted_indices"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::transpose"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "gather_qmm::w"], [0, 1, 1, "_CPPv410gather_qmmRK5arrayRK5arrayRK5arrayRKNSt8optionalI5arrayEENSt8optionalI5arrayEENSt8optionalI5arrayEEbiiRKNSt6stringEb14StreamOrDevice", "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, "_CPPv412logcumsumexpRK5arraybb14StreamOrDevice", "logcumsumexp"], [0, 0, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp"], [0, 1, 1, "_CPPv412logcumsumexpRK5arraybb14StreamOrDevice", "logcumsumexp::a"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::a"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::axis"], [0, 1, 1, "_CPPv412logcumsumexpRK5arraybb14StreamOrDevice", "logcumsumexp::inclusive"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::inclusive"], [0, 1, 1, "_CPPv412logcumsumexpRK5arraybb14StreamOrDevice", "logcumsumexp::reverse"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::reverse"], [0, 1, 1, "_CPPv412logcumsumexpRK5arraybb14StreamOrDevice", "logcumsumexp::s"], [0, 1, 1, "_CPPv412logcumsumexpRK5arrayibb14StreamOrDevice", "logcumsumexp::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, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::bits"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::group_size"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::mode"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::s"], [0, 1, 1, "_CPPv48quantizeRK5arrayiiRKNSt6stringE14StreamOrDevice", "quantize::w"], [0, 0, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::biases"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::bits"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::group_size"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::mode"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::s"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::scales"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::transpose"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "quantized_matmul::w"], [0, 1, 1, "_CPPv416quantized_matmul5array5array5arrayNSt8optionalI5arrayEEbiiRKNSt6stringE14StreamOrDevice", "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, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm"], [0, 1, 1, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm::a"], [0, 1, 1, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm::b"], [0, 1, 1, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm::s"], [0, 1, 1, "_CPPv412segmented_mm5array5array5array14StreamOrDevice", "segmented_mm::segments"], [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"], [344, 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"], [86, 5, 1, "", "array_equal"], [87, 5, 1, "", "as_strided"], [88, 5, 1, "", "async_eval"], [89, 5, 1, "", "atleast_1d"], [90, 5, 1, "", "atleast_2d"], [91, 5, 1, "", "atleast_3d"], [92, 5, 1, "", "bitwise_and"], [93, 5, 1, "", "bitwise_invert"], [94, 5, 1, "", "bitwise_or"], [95, 5, 1, "", "bitwise_xor"], [96, 5, 1, "", "block_masked_mm"], [97, 5, 1, "", "broadcast_arrays"], [98, 5, 1, "", "broadcast_to"], [99, 5, 1, "", "ceil"], [100, 5, 1, "", "clear_cache"], [101, 5, 1, "", "clip"], [102, 5, 1, "", "compile"], [103, 5, 1, "", "concatenate"], [104, 5, 1, "", "conj"], [105, 5, 1, "", "conjugate"], [106, 5, 1, "", "contiguous"], [107, 5, 1, "", "conv1d"], [108, 5, 1, "", "conv2d"], [109, 5, 1, "", "conv3d"], [110, 5, 1, "", "conv_general"], [111, 5, 1, "", "conv_transpose1d"], [112, 5, 1, "", "conv_transpose2d"], [113, 5, 1, "", "conv_transpose3d"], [114, 5, 1, "", "convolve"], [115, 5, 1, "", "cos"], [116, 5, 1, "", "cosh"], [118, 5, 1, "", "cummax"], [119, 5, 1, "", "cummin"], [120, 5, 1, "", "cumprod"], [121, 5, 1, "", "cumsum"], [122, 3, 1, "", "custom_function"], [123, 5, 1, "", "default_device"], [124, 5, 1, "", "default_stream"], [125, 5, 1, "", "degrees"], [126, 5, 1, "", "dequantize"], [127, 5, 1, "", "diag"], [128, 5, 1, "", "diagonal"], [129, 5, 1, "", "disable_compile"], [138, 5, 1, "", "divide"], [139, 5, 1, "", "divmod"], [140, 5, 1, "", "einsum"], [141, 5, 1, "", "einsum_path"], [142, 5, 1, "", "enable_compile"], [143, 5, 1, "", "equal"], [144, 5, 1, "", "erf"], [145, 5, 1, "", "erfinv"], [146, 5, 1, "", "eval"], [147, 5, 1, "", "exp"], [148, 5, 1, "", "expand_dims"], [149, 5, 1, "", "expm1"], [150, 5, 1, "", "export_function"], [151, 5, 1, "", "export_to_dot"], [152, 5, 1, "", "exporter"], [153, 5, 1, "", "eye"], [174, 3, 1, "", "finfo"], [175, 5, 1, "", "flatten"], [176, 5, 1, "", "floor"], [177, 5, 1, "", "floor_divide"], [178, 5, 1, "", "full"], [179, 5, 1, "", "gather_mm"], [180, 5, 1, "", "gather_qmm"], [181, 5, 1, "", "get_active_memory"], [182, 5, 1, "", "get_cache_memory"], [183, 5, 1, "", "get_peak_memory"], [184, 5, 1, "", "grad"], [185, 5, 1, "", "greater"], [186, 5, 1, "", "greater_equal"], [187, 5, 1, "", "hadamard_transform"], [188, 5, 1, "", "identity"], [189, 5, 1, "", "imag"], [190, 5, 1, "", "import_function"], [191, 5, 1, "", "inner"], [192, 5, 1, "", "isclose"], [193, 5, 1, "", "isfinite"], [194, 5, 1, "", "isinf"], [195, 5, 1, "", "isnan"], [196, 5, 1, "", "isneginf"], [197, 5, 1, "", "isposinf"], [198, 5, 1, "", "issubdtype"], [199, 5, 1, "", "jvp"], [200, 5, 1, "", "kron"], [201, 5, 1, "", "left_shift"], [202, 5, 1, "", "less"], [203, 5, 1, "", "less_equal"], [221, 5, 1, "", "linspace"], [222, 5, 1, "", "load"], [223, 5, 1, "", "log"], [224, 5, 1, "", "log10"], [225, 5, 1, "", "log1p"], [226, 5, 1, "", "log2"], [227, 5, 1, "", "logaddexp"], [228, 5, 1, "", "logcumsumexp"], [229, 5, 1, "", "logical_and"], [230, 5, 1, "", "logical_not"], [231, 5, 1, "", "logical_or"], [232, 5, 1, "", "logsumexp"], [233, 5, 1, "", "matmul"], [234, 5, 1, "", "max"], [235, 5, 1, "", "maximum"], [236, 5, 1, "", "mean"], [237, 5, 1, "", "meshgrid"], [242, 5, 1, "", "min"], [243, 5, 1, "", "minimum"], [244, 5, 1, "", "moveaxis"], [245, 5, 1, "", "multiply"], [246, 5, 1, "", "nan_to_num"], [247, 5, 1, "", "negative"], [248, 5, 1, "", "new_stream"], [249, 5, 1, "", "not_equal"], [250, 5, 1, "", "ones"], [251, 5, 1, "", "ones_like"], [252, 5, 1, "", "outer"], [253, 5, 1, "", "pad"], [254, 5, 1, "", "partition"], [255, 5, 1, "", "power"], [256, 5, 1, "", "prod"], [257, 5, 1, "", "put_along_axis"], [258, 5, 1, "", "quantize"], [259, 5, 1, "", "quantized_matmul"], [260, 5, 1, "", "radians"], [274, 5, 1, "", "real"], [275, 5, 1, "", "reciprocal"], [276, 5, 1, "", "remainder"], [277, 5, 1, "", "repeat"], [278, 5, 1, "", "reset_peak_memory"], [279, 5, 1, "", "reshape"], [280, 5, 1, "", "right_shift"], [281, 5, 1, "", "roll"], [282, 5, 1, "", "round"], [283, 5, 1, "", "rsqrt"], [284, 5, 1, "", "save"], [285, 5, 1, "", "save_gguf"], [286, 5, 1, "", "save_safetensors"], [287, 5, 1, "", "savez"], [288, 5, 1, "", "savez_compressed"], [289, 5, 1, "", "set_cache_limit"], [290, 5, 1, "", "set_default_device"], [291, 5, 1, "", "set_default_stream"], [292, 5, 1, "", "set_memory_limit"], [293, 5, 1, "", "set_wired_limit"], [294, 5, 1, "", "sigmoid"], [295, 5, 1, "", "sign"], [296, 5, 1, "", "sin"], [297, 5, 1, "", "sinh"], [298, 5, 1, "", "slice"], [299, 5, 1, "", "slice_update"], [300, 5, 1, "", "softmax"], [301, 5, 1, "", "sort"], [302, 5, 1, "", "split"], [303, 5, 1, "", "sqrt"], [304, 5, 1, "", "square"], [305, 5, 1, "", "squeeze"], [306, 5, 1, "", "stack"], [307, 5, 1, "", "std"], [308, 5, 1, "", "stop_gradient"], [309, 5, 1, "", "stream"], [310, 5, 1, "", "subtract"], [311, 5, 1, "", "sum"], [312, 5, 1, "", "swapaxes"], [313, 5, 1, "", "synchronize"], [314, 5, 1, "", "take"], [315, 5, 1, "", "take_along_axis"], [316, 5, 1, "", "tan"], [317, 5, 1, "", "tanh"], [318, 5, 1, "", "tensordot"], [319, 5, 1, "", "tile"], [320, 5, 1, "", "topk"], [321, 5, 1, "", "trace"], [322, 5, 1, "", "transpose"], [323, 5, 1, "", "tri"], [324, 5, 1, "", "tril"], [325, 5, 1, "", "triu"], [326, 5, 1, "", "unflatten"], [327, 5, 1, "", "value_and_grad"], [328, 5, 1, "", "var"], [329, 5, 1, "", "view"], [330, 5, 1, "", "vjp"], [331, 5, 1, "", "vmap"], [332, 5, 1, "", "where"], [333, 5, 1, "", "zeros"], [334, 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": [[344, 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, 6, 1, "", "imag"], [52, 4, 1, "", "item"], [53, 6, 1, "", "itemsize"], [54, 4, 1, "", "log"], [55, 4, 1, "", "log10"], [56, 4, 1, "", "log1p"], [57, 4, 1, "", "log2"], [58, 4, 1, "", "logcumsumexp"], [59, 4, 1, "", "logsumexp"], [60, 4, 1, "", "max"], [61, 4, 1, "", "mean"], [62, 4, 1, "", "min"], [63, 4, 1, "", "moveaxis"], [64, 6, 1, "", "nbytes"], [65, 6, 1, "", "ndim"], [66, 4, 1, "", "prod"], [67, 6, 1, "", "real"], [68, 4, 1, "", "reciprocal"], [69, 4, 1, "", "reshape"], [70, 4, 1, "", "round"], [71, 4, 1, "", "rsqrt"], [72, 6, 1, "", "shape"], [73, 4, 1, "", "sin"], [74, 6, 1, "", "size"], [75, 4, 1, "", "split"], [76, 4, 1, "", "sqrt"], [77, 4, 1, "", "square"], [78, 4, 1, "", "squeeze"], [79, 4, 1, "", "std"], [80, 4, 1, "", "sum"], [81, 4, 1, "", "swapaxes"], [82, 4, 1, "", "tolist"], [83, 4, 1, "", "transpose"], [84, 4, 1, "", "var"], [85, 4, 1, "", "view"]], "mlx.core.cuda": [[117, 5, 1, "", "is_available"]], "mlx.core.custom_function": [[122, 4, 1, "", "__init__"]], "mlx.core.distributed": [[130, 3, 1, "", "Group"], [131, 5, 1, "", "all_gather"], [132, 5, 1, "", "all_sum"], [133, 5, 1, "", "init"], [134, 5, 1, "", "is_available"], [135, 5, 1, "", "recv"], [136, 5, 1, "", "recv_like"], [137, 5, 1, "", "send"]], "mlx.core.distributed.Group": [[130, 4, 1, "", "__init__"]], "mlx.core.fast": [[154, 5, 1, "", "cuda_kernel"], [155, 5, 1, "", "layer_norm"], [156, 5, 1, "", "metal_kernel"], [157, 5, 1, "", "rms_norm"], [158, 5, 1, "", "rope"], [159, 5, 1, "", "scaled_dot_product_attention"]], "mlx.core.fft": [[160, 5, 1, "", "fft"], [161, 5, 1, "", "fft2"], [162, 5, 1, "", "fftn"], [163, 5, 1, "", "fftshift"], [164, 5, 1, "", "ifft"], [165, 5, 1, "", "ifft2"], [166, 5, 1, "", "ifftn"], [167, 5, 1, "", "ifftshift"], [168, 5, 1, "", "irfft"], [169, 5, 1, "", "irfft2"], [170, 5, 1, "", "irfftn"], [171, 5, 1, "", "rfft"], [172, 5, 1, "", "rfft2"], [173, 5, 1, "", "rfftn"]], "mlx.core.finfo": [[174, 4, 1, "", "__init__"]], "mlx.core.linalg": [[204, 5, 1, "", "cholesky"], [205, 5, 1, "", "cholesky_inv"], [206, 5, 1, "", "cross"], [207, 5, 1, "", "eig"], [208, 5, 1, "", "eigh"], [209, 5, 1, "", "eigvals"], [210, 5, 1, "", "eigvalsh"], [211, 5, 1, "", "inv"], [212, 5, 1, "", "lu"], [213, 5, 1, "", "lu_factor"], [214, 5, 1, "", "norm"], [215, 5, 1, "", "pinv"], [216, 5, 1, "", "qr"], [217, 5, 1, "", "solve"], [218, 5, 1, "", "solve_triangular"], [219, 5, 1, "", "svd"], [220, 5, 1, "", "tri_inv"]], "mlx.core.metal": [[238, 5, 1, "", "device_info"], [239, 5, 1, "", "is_available"], [240, 5, 1, "", "start_capture"], [241, 5, 1, "", "stop_capture"]], "mlx.core.random": [[261, 5, 1, "", "bernoulli"], [262, 5, 1, "", "categorical"], [263, 5, 1, "", "gumbel"], [264, 5, 1, "", "key"], [265, 5, 1, "", "laplace"], [266, 5, 1, "", "multivariate_normal"], [267, 5, 1, "", "normal"], [268, 5, 1, "", "permutation"], [269, 5, 1, "", "randint"], [270, 5, 1, "", "seed"], [271, 5, 1, "", "split"], [272, 5, 1, "", "truncated_normal"], [273, 5, 1, "", "uniform"]], "mlx.nn": [[357, 3, 1, "", "ALiBi"], [358, 3, 1, "", "AvgPool1d"], [359, 3, 1, "", "AvgPool2d"], [360, 3, 1, "", "AvgPool3d"], [361, 3, 1, "", "BatchNorm"], [362, 3, 1, "", "CELU"], [363, 3, 1, "", "Conv1d"], [364, 3, 1, "", "Conv2d"], [365, 3, 1, "", "Conv3d"], [366, 3, 1, "", "ConvTranspose1d"], [367, 3, 1, "", "ConvTranspose2d"], [368, 3, 1, "", "ConvTranspose3d"], [369, 3, 1, "", "Dropout"], [370, 3, 1, "", "Dropout2d"], [371, 3, 1, "", "Dropout3d"], [372, 3, 1, "", "ELU"], [373, 3, 1, "", "Embedding"], [374, 3, 1, "", "GELU"], [375, 3, 1, "", "GLU"], [376, 3, 1, "", "GRU"], [377, 3, 1, "", "GroupNorm"], [378, 3, 1, "", "HardShrink"], [379, 3, 1, "", "HardTanh"], [380, 3, 1, "", "Hardswish"], [381, 3, 1, "", "InstanceNorm"], [382, 3, 1, "", "LSTM"], [383, 3, 1, "", "LayerNorm"], [384, 3, 1, "", "LeakyReLU"], [385, 3, 1, "", "Linear"], [386, 3, 1, "", "LogSigmoid"], [387, 3, 1, "", "LogSoftmax"], [388, 3, 1, "", "MaxPool1d"], [389, 3, 1, "", "MaxPool2d"], [390, 3, 1, "", "MaxPool3d"], [391, 3, 1, "", "Mish"], [488, 3, 1, "", "Module"], [412, 3, 1, "", "MultiHeadAttention"], [413, 3, 1, "", "PReLU"], [414, 3, 1, "", "QuantizedEmbedding"], [415, 3, 1, "", "QuantizedLinear"], [416, 3, 1, "", "RMSNorm"], [417, 3, 1, "", "RNN"], [418, 3, 1, "", "ReLU"], [419, 3, 1, "", "ReLU2"], [420, 3, 1, "", "ReLU6"], [421, 3, 1, "", "RoPE"], [422, 3, 1, "", "SELU"], [423, 3, 1, "", "Sequential"], [424, 3, 1, "", "SiLU"], [425, 3, 1, "", "Sigmoid"], [426, 3, 1, "", "SinusoidalPositionalEncoding"], [427, 3, 1, "", "Softmax"], [428, 3, 1, "", "Softmin"], [429, 3, 1, "", "Softplus"], [430, 3, 1, "", "Softshrink"], [431, 3, 1, "", "Softsign"], [432, 3, 1, "", "Step"], [433, 3, 1, "", "Tanh"], [434, 3, 1, "", "Transformer"], [435, 3, 1, "", "Upsample"], [335, 5, 1, "", "average_gradients"], [444, 3, 1, "", "celu"], [445, 3, 1, "", "elu"], [446, 3, 1, "", "gelu"], [447, 3, 1, "", "gelu_approx"], [448, 3, 1, "", "gelu_fast_approx"], [449, 3, 1, "", "glu"], [450, 3, 1, "", "hard_shrink"], [451, 3, 1, "", "hard_tanh"], [452, 3, 1, "", "hardswish"], [453, 3, 1, "", "leaky_relu"], [454, 3, 1, "", "log_sigmoid"], [455, 3, 1, "", "log_softmax"], [470, 3, 1, "", "mish"], [471, 3, 1, "", "prelu"], [336, 5, 1, "", "quantize"], [472, 3, 1, "", "relu"], [473, 3, 1, "", "relu2"], [474, 3, 1, "", "relu6"], [475, 3, 1, "", "selu"], [476, 3, 1, "", "sigmoid"], [477, 3, 1, "", "silu"], [478, 3, 1, "", "softmax"], [479, 3, 1, "", "softmin"], [480, 3, 1, "", "softplus"], [481, 3, 1, "", "softshrink"], [482, 3, 1, "", "step"], [483, 3, 1, "", "tanh"], [337, 5, 1, "", "value_and_grad"]], "mlx.nn.Module": [[392, 4, 1, "", "apply"], [393, 4, 1, "", "apply_to_modules"], [394, 4, 1, "", "children"], [395, 4, 1, "", "eval"], [396, 4, 1, "", "filter_and_map"], [397, 4, 1, "", "freeze"], [398, 4, 1, "", "leaf_modules"], [399, 4, 1, "", "load_weights"], [400, 4, 1, "", "modules"], [401, 4, 1, "", "named_modules"], [402, 4, 1, "", "parameters"], [403, 4, 1, "", "save_weights"], [404, 4, 1, "", "set_dtype"], [405, 6, 1, "", "state"], [406, 4, 1, "", "train"], [407, 4, 1, "", "trainable_parameters"], [408, 6, 1, "", "training"], [409, 4, 1, "", "unfreeze"], [410, 4, 1, "", "update"], [411, 4, 1, "", "update_modules"]], "mlx.nn.init": [[436, 5, 1, "", "constant"], [437, 5, 1, "", "glorot_normal"], [438, 5, 1, "", "glorot_uniform"], [439, 5, 1, "", "he_normal"], [440, 5, 1, "", "he_uniform"], [441, 5, 1, "", "identity"], [442, 5, 1, "", "normal"], [443, 5, 1, "", "uniform"]], "mlx.nn.losses": [[456, 3, 1, "", "binary_cross_entropy"], [457, 3, 1, "", "cosine_similarity_loss"], [458, 3, 1, "", "cross_entropy"], [459, 3, 1, "", "gaussian_nll_loss"], [460, 3, 1, "", "hinge_loss"], [461, 3, 1, "", "huber_loss"], [462, 3, 1, "", "kl_div_loss"], [463, 3, 1, "", "l1_loss"], [464, 3, 1, "", "log_cosh_loss"], [465, 3, 1, "", "margin_ranking_loss"], [466, 3, 1, "", "mse_loss"], [467, 3, 1, "", "nll_loss"], [468, 3, 1, "", "smooth_l1_loss"], [469, 3, 1, "", "triplet_loss"]], "mlx.optimizers": [[491, 3, 1, "", "AdaDelta"], [492, 3, 1, "", "Adafactor"], [493, 3, 1, "", "Adagrad"], [494, 3, 1, "", "Adam"], [495, 3, 1, "", "AdamW"], [496, 3, 1, "", "Adamax"], [497, 3, 1, "", "Lion"], [498, 3, 1, "", "MultiOptimizer"], [499, 3, 1, "", "Muon"], [512, 3, 1, "", "Optimizer"], [504, 3, 1, "", "RMSprop"], [505, 3, 1, "", "SGD"], [338, 5, 1, "", "clip_grad_norm"], [506, 5, 1, "", "cosine_decay"], [507, 5, 1, "", "exponential_decay"], [508, 5, 1, "", "join_schedules"], [509, 5, 1, "", "linear_schedule"], [510, 5, 1, "", "step_decay"]], "mlx.optimizers.Optimizer": [[500, 4, 1, "", "apply_gradients"], [501, 4, 1, "", "init"], [502, 6, 1, "", "state"], [503, 4, 1, "", "update"]], "mlx.utils": [[339, 5, 1, "", "tree_flatten"], [340, 5, 1, "", "tree_map"], [341, 5, 1, "", "tree_map_with_path"], [342, 5, 1, "", "tree_reduce"], [343, 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, 53, 65, 102, 124, 126, 161, 162, 165, 166, 169, 170, 172, 173, 184, 205, 214, 219, 222, 236, 252, 258, 262, 282, 285, 286, 307, 309, 327, 328, 329, 331, 337, 356, 359, 360, 376, 382, 389, 390, 396, 397, 399, 403, 404, 405, 409, 410, 411, 417, 490, 501, 502, 514, 517, 519, 520, 524, 525, 526, 527], "0": [0, 1, 2, 4, 5, 6, 7, 9, 10, 15, 19, 39, 46, 47, 50, 70, 75, 79, 84, 87, 88, 100, 103, 107, 108, 109, 110, 111, 112, 113, 122, 127, 128, 153, 154, 156, 159, 175, 179, 184, 190, 200, 207, 208, 211, 212, 214, 216, 220, 246, 253, 261, 265, 267, 268, 273, 277, 282, 289, 293, 298, 299, 302, 306, 307, 321, 323, 324, 325, 326, 327, 328, 331, 335, 338, 339, 341, 342, 356, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 377, 378, 381, 383, 384, 388, 389, 390, 413, 418, 421, 426, 430, 432, 434, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 447, 448, 450, 451, 452, 453, 456, 458, 460, 461, 465, 468, 469, 471, 472, 473, 474, 475, 481, 482, 485, 488, 491, 492, 494, 495, 496, 497, 499, 501, 504, 505, 506, 507, 508, 509, 510, 514, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526], "00005": 5, "0001": 426, "0005": 447, "001": 492, "00364": 5, "01": [5, 384, 453, 495, 499], "0137595": 439, "015": 448, "0184009": 440, "02264": 438, "024": 520, "02765": 439, "0300242": 440, "044715": [374, 447], "0485873": 458, "05": [17, 192, 361, 377, 381, 383, 416], "0507": 475, "05202": 6, "06": [459, 469, 491], "0638": 465, "06450": 383, "0645099": 442, "06561": 507, "06675": 497, "07467": 416, "08": [17, 192, 457, 493, 494, 495, 496, 504], "08022": 381, "081": 510, "08415": 448, "08494": 377, "08619": 440, "08681": [391, 470], "09864": 6, "0999938": 508, "0999961": 506, "0d": 499, "0f": 0, "0j": [207, 209], "1": [0, 1, 2, 3, 4, 6, 7, 9, 15, 19, 29, 30, 39, 47, 50, 88, 107, 108, 109, 110, 111, 112, 113, 122, 127, 128, 149, 150, 151, 152, 154, 156, 159, 160, 161, 164, 165, 168, 169, 170, 171, 172, 173, 175, 187, 191, 198, 200, 205, 206, 207, 208, 209, 210, 212, 214, 216, 233, 237, 252, 254, 258, 262, 265, 266, 267, 273, 292, 294, 298, 299, 301, 314, 320, 321, 326, 327, 338, 339, 341, 342, 347, 356, 358, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 375, 376, 377, 381, 382, 383, 385, 388, 413, 416, 417, 421, 425, 426, 432, 435, 437, 438, 439, 440, 441, 442, 443, 444, 445, 447, 448, 449, 451, 454, 455, 456, 457, 458, 459, 460, 461, 462, 464, 465, 467, 468, 469, 475, 476, 478, 479, 480, 482, 485, 488, 490, 491, 492, 493, 494, 495, 496, 497, 501, 504, 505, 506, 507, 508, 509, 510, 517, 518, 519, 520, 521, 522, 524, 525, 526, 527], "10": [0, 3, 6, 7, 200, 224, 282, 287, 340, 356, 399, 485, 508, 510, 517, 518, 521], "100": [2, 5, 6, 456, 509, 517, 520, 523, 527], "1000": [2, 159, 506, 517], "10000": 421, "101": 509, "1024": [1, 6], "105361": 456, "10_000": 5, "10x": 497, "11": 214, "12": [6, 9, 187, 200, 508], "1212": 491, "123": [518, 522], "12451": 438, "128": [159, 287, 356], "13": 9, "14": [9, 200], "15": [1, 9, 200, 214, 293, 342, 517], "150594": 437, "15268": 439, "16": [1, 154, 156, 347, 358, 360, 381, 388, 390, 392, 488], "1606": 448, "1607": [381, 383], "16384": 187, "16506": 440, "168": 518, "17": [4, 9], "177208": 439, "18": 200, "1803": 377, "1908": [391, 470], "1910": 416, "191107": 437, "192": 518, "1985": 214, "1_000": 5, "1_all": 9, "1d": [0, 107, 111, 114, 285, 315, 499], "1e": [0, 5, 7, 17, 192, 361, 377, 381, 383, 384, 416, 457, 459, 469, 490, 491, 492, 493, 494, 495, 496, 501, 504, 506, 507, 508, 509, 510], "1e3": 517, "1st": 258, "2": [0, 1, 2, 4, 5, 6, 7, 9, 39, 108, 112, 122, 127, 128, 144, 150, 151, 152, 159, 161, 165, 168, 169, 170, 171, 172, 173, 175, 187, 198, 200, 204, 205, 206, 207, 208, 209, 210, 211, 212, 214, 215, 216, 219, 220, 226, 233, 258, 266, 267, 271, 298, 299, 318, 321, 323, 324, 325, 326, 338, 342, 347, 356, 358, 359, 360, 364, 367, 374, 384, 388, 389, 390, 416, 426, 435, 436, 437, 438, 439, 440, 441, 442, 443, 447, 458, 459, 461, 468, 469, 473, 485, 488, 490, 491, 493, 494, 495, 501, 504, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "20": [187, 200, 214], "200": [6, 508, 520], "2002": 6, "2011": 493, "2012": [491, 504], "2015": [370, 494, 496], "2019": [6, 495], "2020": 6, "2021": 6, "20397": 456, "20_000": 6, "21": [6, 200, 510], "2104": 6, "223144": 456, "223404": 438, "225": 214, "225763": 465, "2302": 497, "23607": [214, 216], "24": 200, "24264": 214, "247": 6, "25": [9, 413, 435], "25211": 439, "256": [1, 7, 154, 156], "256995": 465, "27": 4, "28": [187, 200], "2d": [0, 108, 112, 128, 258, 361, 370], "2nd": 258, "2x": 524, "3": [0, 1, 2, 4, 6, 9, 88, 109, 113, 122, 150, 152, 175, 198, 200, 206, 207, 208, 209, 210, 214, 216, 298, 299, 326, 338, 342, 360, 365, 368, 374, 390, 435, 438, 440, 447, 452, 492, 497, 514, 517, 518, 519, 521, 524, 525], "30": 492, "3118": 524, "32": [1, 6, 7, 96, 159, 258, 259, 347, 359, 360, 389, 390, 416, 517], "32mib": 335, "330": 6, "33333": 435, "33554432": 335, "348587": 458, "35": 9, "363207": 437, "36788": 517, "379159": 438, "380709": 442, "39": 6, "3d": [0, 2, 109, 113, 158, 361, 371, 435], "3f": [2, 7, 517], "3x": 2, "4": [0, 1, 2, 6, 126, 156, 159, 175, 180, 200, 214, 258, 259, 287, 298, 326, 336, 342, 347, 358, 359, 360, 361, 381, 388, 389, 390, 414, 415, 434, 435, 437, 438, 439, 456, 517, 518, 519, 521, 525, 527], "4096": [2, 517, 520, 527], "40x": 1, "41421": 214, "417497": 443, "42": 343, "437": 6, "44": 6, "447214": 216, "458835": 439, "475": 6, "48095": 437, "4d": [1, 435, 499], "4m": 1, "5": [0, 1, 2, 5, 6, 9, 159, 200, 214, 261, 292, 298, 342, 358, 361, 369, 370, 371, 374, 378, 381, 388, 430, 435, 436, 439, 440, 447, 450, 468, 481, 485, 490, 499, 504, 506, 507, 517, 520, 521], "50": [0, 221], "500": [6, 527], "510826": 456, "512": [3, 6, 434, 527], "534422": 442, "539245": 456, "53947": 437, "54": 9, "55": 1, "550": 9, "559": 2, "5701": 491, "573409": 465, "57771": 216, "579": 6, "5f": 5, "6": [1, 2, 6, 122, 200, 214, 287, 298, 420, 434, 438, 447, 448, 452, 459, 469, 474, 504, 517, 521, 525], "61278": 437, "617261": 443, "628": 6, "633": 6, "639": 520, "64": [0, 1, 96, 126, 180, 258, 259, 336, 347, 414, 415], "64331": 440, "666329": 440, "66667": 435, "67326": 475, "676": 1, "690": 6, "6967": 439, "7": [2, 6, 9, 200, 214, 258, 521], "702": [374, 448], "707107": [207, 208], "71828": 517, "74166": 214, "74597": 214, "75": 435, "75596": 465, "75787": 439, "765166": 465, "773433": 465, "774": 2, "776856": 438, "793615": 440, "79854": 440, "7b": 6, "7m": 1, "8": [0, 1, 2, 6, 214, 258, 347, 359, 360, 381, 389, 390, 434, 457, 491, 492, 493, 494, 495, 496, 504, 517, 521, 525, 527], "8192": [6, 187], "84804": 214, "863726": 443, "883935": 443, "890597": 438, "894427": 216, "89613": 437, "8gb": 6, "8x": 1, "9": [4, 9, 214, 458, 491, 494, 495, 496, 497, 501, 507, 510, 524], "90041": 438, "912766": 438, "916291": 456, "95": [7, 499], "982273": 442, "99": [497, 504], "995016": 437, "999": [494, 495, 496], "A": [0, 2, 6, 8, 9, 10, 72, 86, 102, 150, 151, 154, 155, 156, 157, 159, 184, 198, 199, 205, 207, 208, 209, 210, 212, 214, 216, 219, 222, 232, 233, 234, 238, 242, 258, 261, 262, 263, 265, 266, 267, 268, 269, 272, 273, 302, 306, 309, 327, 330, 331, 336, 337, 338, 339, 340, 341, 342, 343, 344, 356, 361, 370, 376, 377, 381, 383, 396, 400, 401, 404, 410, 411, 416, 423, 426, 434, 437, 438, 440, 448, 469, 470, 488, 490, 494, 496, 498, 500, 501, 503, 508, 517, 518, 519, 520, 522, 523, 524], "AS": 179, "And": [4, 6, 435], "As": [7, 39, 314, 356, 518], "At": [101, 326, 518], "But": [519, 527], "By": [6, 336, 404, 456, 518, 520, 524], "For": [0, 1, 2, 4, 6, 9, 39, 122, 126, 159, 179, 198, 214, 258, 343, 356, 361, 370, 374, 392, 397, 406, 409, 415, 421, 426, 435, 437, 438, 439, 440, 456, 485, 490, 499, 514, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "If": [0, 1, 2, 4, 6, 9, 16, 17, 18, 19, 27, 28, 29, 30, 82, 86, 87, 88, 101, 103, 114, 118, 119, 120, 121, 127, 128, 131, 132, 133, 135, 136, 137, 146, 155, 157, 158, 159, 163, 167, 171, 172, 173, 177, 178, 179, 184, 192, 204, 205, 206, 214, 219, 222, 228, 232, 233, 234, 236, 237, 242, 246, 250, 253, 254, 256, 257, 262, 266, 267, 268, 277, 281, 289, 292, 300, 301, 302, 307, 311, 313, 314, 315, 318, 320, 321, 327, 328, 331, 333, 335, 336, 339, 340, 342, 361, 363, 364, 365, 366, 367, 368, 377, 383, 385, 397, 399, 409, 410, 411, 415, 417, 421, 423, 426, 435, 456, 458, 469, 492, 494, 495, 517, 518, 519, 520, 522, 523, 526, 527, 528], "In": [0, 1, 2, 6, 7, 39, 159, 233, 258, 340, 356, 370, 377, 488, 491, 493, 494, 496, 497, 500, 516, 517, 518, 519, 520, 522, 523, 526, 527], "It": [2, 6, 9, 136, 184, 291, 327, 338, 342, 356, 411, 415, 500, 512, 518, 522, 524, 526], "Its": [356, 519], "No": [2, 6, 208, 210, 518], "Not": [102, 249, 517], "ON": [3, 4, 9], "Of": 520, "On": [1, 517, 520, 523], "One": [160, 164, 171, 253, 283, 517, 519, 520, 522], "THE": 9, "That": [6, 267], "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, 51, 53, 64, 65, 67, 72, 82, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 125, 126, 127, 128, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 143, 144, 145, 147, 148, 149, 150, 151, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 177, 178, 179, 180, 182, 183, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 199, 200, 201, 202, 203, 206, 207, 208, 209, 210, 212, 213, 214, 216, 217, 218, 219, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 240, 242, 243, 244, 245, 247, 249, 250, 251, 252, 253, 254, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 285, 286, 289, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 347, 349, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 369, 370, 371, 373, 375, 376, 377, 381, 382, 383, 385, 388, 389, 390, 392, 393, 397, 399, 403, 404, 405, 406, 409, 410, 411, 412, 414, 415, 416, 417, 421, 423, 426, 432, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 482, 485, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 502, 504, 505, 506, 509, 512, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528], "Then": [5, 9], "There": [1, 356, 435, 517], "These": [1, 2, 102, 257, 315, 458, 527], "To": [0, 1, 2, 3, 5, 6, 7, 9, 212, 258, 289, 356, 485, 490, 517, 518, 519, 520, 525], "With": 519, "_": [1, 3, 5, 6, 341, 356, 506, 507, 508, 509, 510, 514, 517, 523, 527], "__call__": [1, 6, 7, 356, 488, 519], "__init__": [2, 6, 7, 10, 11, 12, 31, 122, 130, 174, 344, 356, 488], "__main__": [2, 6], "__name__": [2, 6], "_a": 2, "_ext": 2, "_f": 214, "_in": [437, 438], "_out": [437, 438], "_p": 469, "_val": 451, "a1": 179, "a2": 179, "a_": 214, "a_max": [0, 101], "a_min": [0, 101], "a_ndim": 1, "a_shap": 1, "a_strid": 1, "a_view": 524, "ab": [0, 17, 192, 214, 327, 377, 381, 383, 391, 416, 448, 470, 517, 519], "abil": 518, "abl": [2, 4, 522], "abort": 122, "about": [1, 2, 6, 7, 141, 238, 523, 527], "abov": [1, 2, 6, 258, 324, 356, 435, 518, 519, 520, 521, 522, 523, 527], "absolut": [0, 13, 17, 192, 447, 448, 468, 518], "acc": 342, "acceler": [4, 361], "accept": [518, 522], "access": [0, 6, 52, 356, 488, 501, 518, 523, 527], "accord": [0, 263, 332, 336, 412, 437, 438, 439, 440], "accordingli": 2, "accumul": [342, 416], "accuraci": 7, "accustom": 6, "achiev": [356, 518], "across": [1, 2, 9, 335, 377, 518], "act": [2, 464], "action": 356, "activ": [2, 9, 181, 370, 419, 432, 434, 450, 470, 473, 481, 482, 484, 517], "actual": [6, 19, 399, 488, 523], "ad": [0, 1, 2, 5, 9, 155, 341, 366, 367, 368, 381, 488, 491, 492, 493, 494, 495, 496, 504, 518, 523, 526], "adadelta": 490, "adafactor": 490, "adagrad": 490, "adam": [490, 496, 497, 508, 509], "adamax": 490, "adamw": [490, 497, 499], "adapt": [491, 492, 493, 518], "add": [0, 1, 2, 3, 4, 6, 15, 39, 148, 227, 253, 363, 364, 365, 366, 367, 368, 519, 520, 522, 527], "add_argu": 6, "add_depend": 2, "add_execut": 4, "add_fun": 519, "add_librari": 2, "addit": [0, 2, 4, 6, 9, 14, 15, 150, 155, 157, 159, 222, 361, 366, 367, 368, 377, 383, 412, 416, 488, 520], "addmm": 0, "address": 2, "adjac": 370, "advanc": [6, 517], "advantag": 527, "advis": 524, "affin": [0, 126, 180, 258, 259, 336, 361, 377, 381, 383, 385, 414, 415], "after": [2, 6, 7, 29, 100, 175, 177, 180, 254, 258, 361, 377, 383, 392, 393, 397, 399, 406, 409, 410, 411, 412, 434, 468, 517, 518, 527], "after_1": 253, "after_2": 253, "after_i": 253, "after_n": 253, "afternoon": 6, "again": [6, 9, 356, 517], "against": [0, 4, 97], "aggreg": [412, 518], "ago": 6, "ai": 122, "aim": 518, "ainv": [211, 220], "albeit": 527, "algebra": 8, "algorithm": [435, 497], "alia": [104, 105, 374], "alibi": 356, "align": [205, 258, 376, 382], "align_corn": 435, "all": [0, 1, 2, 3, 7, 9, 17, 29, 39, 89, 90, 91, 102, 108, 109, 110, 112, 113, 122, 131, 132, 133, 151, 153, 162, 163, 166, 167, 170, 173, 179, 180, 207, 209, 219, 233, 253, 254, 281, 305, 335, 336, 356, 392, 393, 397, 400, 401, 402, 407, 409, 412, 426, 434, 435, 485, 488, 512, 514, 517, 521, 522, 523, 525, 528], "all_avg": 518, "all_reduce_grad": 518, "all_reduce_s": 335, "all_sum": 518, "allclos": [0, 1, 154, 156], "alloc": [2, 182, 289, 292, 488], "allow": [0, 1, 2, 150, 152, 179, 180, 198, 338, 356, 411, 488, 512, 518, 521, 522, 525], "allow_col_major": [0, 106], "almost": [6, 518], "alon": [2, 524], "along": [0, 2, 27, 28, 102, 103, 118, 119, 120, 121, 131, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 179, 180, 187, 206, 214, 228, 257, 268, 277, 281, 300, 302, 306, 314, 315, 318, 319, 320, 321, 329, 356, 375, 417, 449], "alpha": [0, 2, 15, 258, 362, 372, 444, 445, 469, 471, 475, 495, 504], "alpha_": 2, "alreadi": [2, 3, 6, 518], "also": [0, 1, 2, 4, 6, 7, 8, 9, 12, 14, 92, 94, 95, 129, 138, 139, 143, 162, 166, 170, 173, 185, 186, 201, 202, 203, 227, 235, 243, 245, 249, 255, 258, 276, 280, 310, 336, 337, 349, 356, 396, 410, 412, 414, 415, 424, 446, 475, 477, 484, 490, 517, 518, 519, 520, 521, 522, 523, 524, 525, 528], "altern": 514, "although": 518, "alwai": [1, 87, 181, 190, 207, 209, 339, 518, 519, 520], "am": 6, "among": 2, "amount": [6, 183, 292, 358, 388, 519], "amus": 6, "an": [0, 1, 2, 3, 4, 6, 7, 9, 11, 16, 18, 31, 88, 89, 90, 91, 98, 106, 107, 108, 109, 110, 111, 112, 113, 130, 135, 136, 137, 146, 150, 152, 153, 155, 158, 159, 175, 178, 188, 190, 193, 204, 214, 222, 244, 250, 251, 253, 256, 257, 258, 259, 268, 277, 279, 281, 282, 292, 293, 302, 305, 312, 314, 315, 318, 319, 323, 326, 331, 333, 334, 339, 340, 341, 342, 347, 356, 369, 374, 377, 382, 383, 385, 392, 412, 413, 415, 417, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 447, 471, 485, 490, 491, 499, 503, 507, 512, 514, 516, 517, 518, 519, 520, 521, 523, 524, 525, 526, 527, 528], "anaconda": 518, "anchor": 469, "angl": [125, 260, 384], "angular": [158, 421], "ani": [0, 1, 2, 6, 8, 19, 102, 122, 133, 335, 339, 340, 341, 342, 343, 356, 374, 392, 393, 396, 405, 415, 434, 435, 485, 499, 509, 516, 517, 518, 520, 523, 525, 526, 527], "anonym": 517, "anoth": [0, 97, 101, 198, 233, 310, 332, 347, 356, 392, 517, 519, 520, 521, 527], "anwywher": 9, "anyhow": 6, "anymor": 6, "anyth": [6, 327, 518, 523], "anytim": 523, "api": [1, 2, 88, 150, 152, 190, 374, 518, 519, 520], "aplu": 215, "app": 9, "append": [6, 233, 517, 523], "appl": [2, 6, 8, 9, 527], "appli": [0, 39, 158, 159, 179, 219, 340, 341, 342, 356, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 370, 371, 372, 374, 375, 377, 378, 379, 380, 381, 383, 384, 385, 386, 387, 388, 389, 390, 391, 393, 406, 413, 415, 416, 417, 418, 419, 420, 422, 424, 425, 427, 428, 429, 430, 431, 432, 433, 435, 444, 445, 446, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 485, 494, 495, 500, 503, 509, 512, 517, 518], "applic": [3, 9], "apply_fn": 393, "apply_gradi": 490, "apply_to_modul": [356, 397], "approach": [464, 520], "appropri": 517, "approx": 374, "approxim": [17, 374, 446, 447, 448], "apt": 9, "ar": [0, 1, 2, 5, 6, 7, 8, 9, 17, 19, 86, 88, 96, 97, 98, 101, 102, 110, 114, 122, 126, 128, 133, 135, 136, 146, 153, 154, 156, 159, 161, 162, 165, 166, 169, 170, 172, 173, 175, 179, 180, 184, 192, 193, 194, 195, 196, 197, 198, 199, 207, 208, 209, 210, 212, 214, 216, 222, 233, 252, 253, 254, 258, 259, 261, 262, 263, 267, 268, 269, 272, 273, 281, 287, 288, 305, 306, 314, 327, 330, 331, 336, 339, 340, 347, 361, 363, 364, 365, 366, 367, 368, 369, 370, 371, 377, 381, 383, 385, 399, 412, 415, 435, 456, 458, 459, 484, 488, 490, 497, 501, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "arang": [0, 1, 214, 268, 347, 435, 521, 524], "arbitrari": [339, 488, 518], "arbitrarili": [1, 102, 356, 516, 520, 525], "arc": 0, "arcco": 0, "arccosh": 0, "architectur": [6, 9, 238, 356, 411, 527], "archiv": 526, "arcsin": 0, "arcsinh": 0, "arctan": 0, "arctan2": 0, "arctanh": 0, "arg": [2, 6, 11, 19, 88, 130, 146, 150, 151, 152, 174, 190, 287, 288, 344, 522], "arg0": 150, "arg1": 198, "arg2": 198, "argmax": [0, 7], "argmin": 0, "argnam": [184, 327], "argnum": [2, 122, 184, 327, 520], "argpars": 6, "argpartit": 0, "argsort": 0, "argument": [1, 32, 69, 83, 88, 102, 146, 184, 327, 340, 341, 342, 356, 435, 514, 518, 519, 520, 522, 526, 527, 528], "argumentpars": 6, "ari": [89, 90, 91], "aris": 524, "arm": 9, "arm64": 9, "around": 6, "arr": [0, 284, 521], "arr_0": 526, "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, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 122, 125, 126, 127, 128, 131, 132, 135, 136, 137, 138, 139, 140, 141, 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, 175, 176, 177, 178, 179, 180, 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, 242, 243, 244, 245, 246, 247, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 338, 347, 356, 361, 382, 392, 399, 402, 407, 413, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 482, 485, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 504, 505, 506, 507, 508, 509, 510, 517, 518, 519, 520, 523, 524, 525, 527], "array_equ": [0, 17, 192], "arrayfir": 8, "arxiv": [6, 377, 381, 383, 391, 416, 448, 470, 491, 497], "as_strid": 0, "ascend": [208, 210], "ask": [6, 518, 522], "assert": [1, 2, 154, 156], "assign": [0, 2, 39, 488, 518], "associ": [2, 287, 288, 523], "assum": [0, 6, 96, 206, 208, 210, 216, 340, 356, 377, 518], "ast": 1, "astyp": [0, 1, 2, 6, 154, 156, 392, 524], "asynchron": 88, "atleast": 0, "atleast_1d": 0, "atleast_2d": 0, "atleast_3d": 0, "atol": [0, 17, 192], "atom": [1, 156], "atomic_fetch_add_explicit": 1, "atomic_output": [1, 156], "attach": 2, "attempt": [102, 518], "attend": 412, "attent": [159, 397, 412, 426, 434], "attention_norm": 6, "attribut": [1, 10, 11, 12, 31, 174, 344, 405, 488, 512], "audio": 435, "auto": [0, 2, 4, 9, 154, 518, 519], "autom": 520, "automat": [1, 2, 8, 154, 156, 222, 518, 525, 526, 527], "autoregress": 6, "avail": [2, 5, 6, 7, 9, 11, 117, 133, 134, 239, 292, 349, 518, 522, 527], "averag": [335, 358, 359, 360, 491, 492, 494, 495, 496, 518], "avgpool1d": 356, "avgpool2d": 356, "avgpool3d": 356, "avoid": [1, 2, 404, 517, 518], "awai": [2, 6], "awar": [517, 523], "ax": [0, 2, 16, 18, 27, 28, 83, 122, 148, 161, 162, 163, 165, 166, 167, 169, 170, 172, 173, 175, 191, 214, 217, 218, 232, 234, 236, 242, 253, 256, 281, 298, 299, 300, 305, 307, 311, 312, 318, 322, 328, 520], "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, 58, 59, 60, 61, 62, 66, 75, 78, 79, 80, 84, 103, 118, 119, 120, 121, 128, 131, 148, 155, 157, 160, 164, 168, 169, 170, 171, 172, 173, 175, 187, 206, 212, 214, 228, 232, 234, 236, 242, 244, 253, 254, 256, 257, 262, 268, 277, 281, 300, 301, 302, 305, 306, 307, 311, 312, 314, 315, 319, 320, 321, 322, 326, 328, 329, 331, 358, 359, 360, 375, 388, 389, 390, 417, 449, 455, 457, 458, 462, 467, 469, 478, 479, 521], "axis1": [0, 47, 81, 128, 312, 321], "axis2": [0, 47, 81, 128, 312, 321], "axpbi": 2, "axpby_gener": 2, "axpby_general_": 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, 86, 92, 94, 95, 96, 138, 139, 143, 154, 156, 158, 159, 177, 179, 185, 186, 190, 191, 192, 200, 201, 202, 203, 206, 214, 217, 218, 227, 229, 231, 233, 235, 243, 245, 249, 252, 255, 258, 265, 276, 280, 310, 318, 327, 339, 341, 342, 375, 385, 417, 435, 449, 520, 521, 523, 524, 525, 526, 527], "b1": 179, "b2": 179, "b_": [376, 382], "b_stride": 1, "ba": [494, 496], "back": [6, 117, 122, 239, 335, 524], "backend": [1, 9, 133, 134, 522], "backward": [1, 517, 520], "bad": 523, "balanc": 464, "baltimor": 214, "bandwidth": [517, 518], "base": [0, 2, 4, 158, 224, 226, 255, 421, 434, 488, 490, 496, 512, 514, 517, 521], "base_idx": 1, "basi": 512, "basic": [5, 282, 520], "batch": [6, 15, 96, 158, 159, 179, 180, 233, 266, 361, 363, 364, 365, 366, 367, 368, 370, 371, 376, 382, 412, 417, 435, 523], "batch_idx": 1, "batch_iter": [7, 490], "batch_siz": [7, 490], "batchnorm": 356, "becaus": [6, 181, 356, 517, 518, 519, 523], "becom": 133, "been": [0, 2, 6, 182, 523], "befor": [1, 2, 6, 9, 29, 154, 156, 254, 335, 396, 434, 501, 518, 521, 523], "before_1": 253, "before_2": 253, "before_i": 253, "before_n": 253, "beforehand": 252, "beggin": 281, "begin": [87, 183, 205, 258, 376, 382, 432, 450, 461, 468, 475, 481, 482, 518], "behav": 122, "behavior": [167, 212, 266, 464, 521, 523], "behaviour": [122, 204, 205], "behind": 520, "being": [308, 356], "bell": 2, "below": [2, 9, 214, 258, 323, 325, 347, 435, 518, 523], "bench": 2, "benchmark": [2, 517], "benefici": [370, 371, 523], "benefit": 518, "best": 518, "beta": [0, 2, 15, 126, 258, 361, 377, 381, 383, 468, 490, 494, 495, 496, 497], "beta_": 2, "beta_1": [492, 494, 495, 496, 497], "beta_2": [494, 495, 496, 497], "better": [335, 499, 520, 527], "between": [0, 2, 8, 101, 175, 434, 457, 460, 461, 464, 508, 518, 522, 523, 524, 527], "beyond": [281, 506, 509], "bfloat16": [2, 12, 187, 347, 524], "bfloat16_t": 2, "bia": [6, 126, 155, 180, 258, 259, 340, 356, 363, 364, 365, 366, 367, 368, 376, 382, 383, 385, 397, 399, 409, 412, 415, 417, 494, 495, 496, 501, 520], "bias": [0, 126, 180, 258, 259, 376, 382, 397, 409, 412], "bias_correct": [494, 495], "bicub": 435, "big": [1, 335, 517], "bigger": [6, 492], "bilinear": [1, 435], "binari": [222, 284, 285, 286, 287, 288, 329, 432, 456, 482, 517, 522], "binary_cross_entropi": [356, 517], "bind": 522, "bit": [0, 126, 180, 201, 258, 259, 280, 336, 347, 392, 414, 415, 416], "bitwis": [0, 92, 93, 94, 95, 201, 280], "bitwise_and": 0, "bitwise_invert": 0, "bitwise_or": 0, "bitwise_xor": 0, "bla": 9, "block": [0, 2, 6, 96, 434], "block_masked_mm": 0, "block_siz": [0, 96], "bn": 361, "bodi": [1, 154, 156], "bool": [0, 1, 2, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 58, 59, 60, 61, 62, 66, 79, 80, 82, 84, 86, 102, 106, 110, 117, 118, 119, 120, 121, 133, 134, 150, 152, 154, 156, 158, 179, 180, 192, 198, 204, 205, 214, 218, 219, 220, 222, 228, 232, 234, 236, 237, 239, 242, 256, 259, 307, 311, 328, 336, 361, 363, 364, 365, 366, 367, 368, 376, 377, 381, 382, 383, 385, 392, 396, 397, 399, 404, 406, 409, 410, 411, 412, 415, 417, 421, 426, 434, 435, 456, 459, 492, 494, 495, 498, 499, 505], "bool_": [12, 347], "boolean": [0, 17, 86, 159, 192, 193, 194, 195, 196, 197, 198, 229, 230, 231, 347, 408, 521], "both": [1, 2, 14, 92, 94, 95, 138, 139, 143, 185, 186, 198, 201, 202, 203, 214, 227, 235, 243, 245, 249, 255, 262, 276, 280, 310, 336, 358, 359, 360, 381, 382, 388, 389, 390, 490, 517, 518, 519, 520, 525, 527], "bottom": 435, "bound": [0, 269, 272, 273, 374, 443, 517, 521, 527], "boundari": 508, "bracket": 6, "brain": 347, "break": 524, "bregler": 370, "bridg": 518, "broadcast": [0, 2, 14, 17, 92, 94, 95, 97, 98, 101, 138, 139, 143, 159, 178, 185, 186, 192, 201, 202, 203, 227, 233, 235, 243, 245, 249, 255, 257, 261, 262, 266, 269, 272, 273, 276, 280, 310, 315, 332, 412], "broadcast_arrai": [0, 2], "broadcast_to": 0, "broadcasted_input": 2, "brought": 8, "btl_tcp_if_includ": [518, 522], "btl_tcp_link": [518, 522], "buffer": [1, 2, 181, 524], "bui": 6, "build": [1, 3, 4, 6, 8, 439, 488, 517, 519], "build_ext": [2, 9], "build_shared_lib": [2, 9], "built": [1, 2, 4, 9, 523], "bundl": 6, "byte": [53, 64, 181, 182, 183, 289, 292, 293, 335, 347, 522], "c": [0, 1, 2, 6, 15, 214, 361, 363, 364, 365, 366, 367, 368, 370, 371, 381, 382, 524, 525, 527], "c_": [382, 497], "c_in": [107, 108, 109, 110, 111, 112, 113], "c_out": [107, 108, 109, 110, 111, 112, 113], "c_pad": 1, "c_t": [382, 497], "cabl": 518, "cach": [6, 9, 100, 181, 182, 289, 517], "calcul": [214, 215, 456, 459, 465, 492], "call": [2, 3, 6, 7, 32, 100, 133, 136, 177, 183, 190, 335, 356, 373, 397, 409, 414, 423, 488, 490, 501, 517, 518, 519, 520, 522, 523], "callabl": [102, 122, 150, 152, 154, 156, 184, 190, 199, 327, 330, 331, 336, 337, 339, 340, 341, 342, 392, 393, 396, 404, 417, 423, 434, 436, 437, 438, 439, 440, 441, 442, 443, 491, 492, 493, 494, 495, 496, 497, 498, 499, 504, 505, 506, 507, 508, 509, 510], "callback": 150, "can": [1, 2, 3, 4, 6, 8, 9, 14, 19, 69, 83, 87, 88, 92, 94, 95, 102, 128, 129, 130, 138, 139, 143, 146, 150, 151, 158, 159, 179, 185, 186, 190, 201, 202, 203, 214, 227, 235, 243, 245, 249, 255, 258, 261, 262, 269, 272, 273, 276, 280, 285, 293, 310, 321, 326, 327, 335, 342, 356, 359, 360, 373, 374, 389, 390, 396, 409, 414, 423, 435, 458, 485, 488, 490, 500, 501, 514, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528], "cannot": [6, 101, 521, 524], "captur": [2, 3, 102, 122, 240, 241, 356, 517], "care": [6, 518, 519, 522, 523], "carefulli": [517, 519], "carri": 2, "cartesian": 237, "case": [2, 6, 131, 132, 133, 135, 136, 137, 162, 166, 168, 170, 171, 172, 173, 175, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 215, 216, 217, 218, 219, 220, 233, 279, 305, 326, 359, 360, 370, 389, 390, 432, 450, 468, 475, 481, 482, 500, 501, 517, 518, 519, 520, 522, 525, 526, 527, 528], "cast": [2, 38, 171, 172, 173, 222, 335, 392, 404, 524], "caster": 2, "categor": 6, "categori": [12, 198, 347], "caus": [356, 517, 523], "causal": [6, 159], "caution": 87, "cd": [3, 9], "cdf": [263, 374, 446], "cdot": [448, 457, 460, 477], "ceil": 0, "ceildiv": 1, "cell": 382, "celu": 356, "center": 163, "certain": [406, 517], "chang": [87, 88, 102, 150, 152, 190, 291, 329, 410, 415, 435, 461, 468, 517, 524], "channel": [1, 107, 108, 109, 110, 111, 112, 113, 361, 363, 364, 365, 366, 367, 368, 370, 371], "channel_idx": 1, "char": 2, "charact": 339, "check": [0, 2, 9, 86, 117, 134, 198, 208, 210, 239, 399, 410, 411, 518, 519, 520, 521], "checklist": [518, 522], "checkout": [3, 517], "checkpoint": [434, 490], "chen": 497, "child": 411, "children": 356, "chip": 9, "choleski": 205, "choos": [6, 158, 421, 522], "chosen": 141, "clamp": 175, "clang": 9, "clarifi": 518, "clariti": 520, "class": [2, 6, 7, 10, 11, 12, 31, 122, 130, 174, 344, 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, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 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, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 504, 505, 512], "class_pred": 336, "classif": [439, 440], "classifi": 7, "classmethod": [414, 415], "clear": 100, "click": 9, "clip": [0, 338, 456, 492], "clip_threshold": 492, "clipped_grad": 338, "clone": 9, "close": [5, 8, 9, 17, 192], "closer": 340, "cmake": [3, 4, 9], "cmake_arg": [3, 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, 122, 426, 520], "code": [1, 154, 156, 517, 518, 519, 523], "coeffici": [2, 491, 492, 494, 495, 496, 497], "col": [106, 323], "cold": 9, "collect": [340, 341, 516], "column": [106, 153, 188, 207, 208, 258], "com": 9, "combin": [6, 219, 342], "come": [2, 6, 518, 520], "command": [2, 3, 4, 9, 518, 522], "command_buff": 2, "common": [490, 517, 523], "commonli": [7, 410, 485, 517], "commun": [8, 130, 133, 134, 335, 522], "communication_stream": 335, "communication_typ": 335, "compact": 213, "compar": [2, 86, 517], "comparison": [17, 143, 185, 186, 202, 203, 249], "compat": [6, 150, 152, 159, 190, 262, 266, 374, 526], "compil": [0, 1, 3, 4, 8, 9, 129, 142, 154, 156, 518, 519, 520, 523], "compiled_fun": [517, 519], "compiled_grad_fn": 517, "complement": 93, "complet": [5, 6, 9, 410, 411, 519, 520, 527], "complex": [2, 51, 67, 104, 105, 169, 170, 171, 172, 173, 189, 207, 208, 209, 210, 267, 274, 339, 347, 356, 411, 517, 519, 520], "complex64": [2, 12, 207, 209, 347], "complex64_t": 2, "complexflo": 12, "compon": [2, 4, 6, 163, 219], "compos": [8, 356, 517, 520, 525], "composit": 525, "compress": 288, "compromis": 6, "comput": [0, 1, 2, 5, 6, 7, 8, 9, 118, 119, 120, 121, 122, 126, 141, 149, 158, 184, 199, 200, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 217, 218, 220, 227, 228, 236, 252, 258, 276, 300, 307, 308, 318, 327, 328, 330, 337, 356, 361, 376, 377, 381, 382, 383, 397, 410, 415, 416, 421, 434, 437, 438, 439, 440, 447, 448, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 490, 491, 492, 494, 495, 496, 497, 503, 517, 518, 519, 520, 525, 527], "computation": 523, "compute_encod": 2, "compute_uv": 219, "concaten": [0, 6, 131, 335], "concept": 488, "concis": 6, "concret": [2, 376, 382, 385, 417, 523, 527], "conda": [9, 518], "condit": [0, 332, 517, 527], "config": [2, 4, 518], "configu": 490, "configur": 518, "confirm": [518, 522], "confus": 7, "conj": 105, "conjug": [0, 104], "connect": [499, 518, 522], "consecut": [158, 258, 421], "consequ": 6, "consid": [6, 17, 86, 106, 192, 339, 340, 341, 377, 498, 516, 518], "consider": 517, "const": [0, 1, 2, 459], "constant": [0, 2, 6, 9, 122, 155, 157, 253, 356, 361, 377, 383, 416, 459, 469, 504, 506, 517, 519, 524], "constant_valu": 253, "constitut": 340, "construct": [0, 2, 7, 46, 127, 178, 212, 250, 319, 333], "consult": 518, "consum": 523, "contain": [2, 6, 9, 29, 30, 72, 102, 128, 141, 168, 169, 170, 179, 180, 207, 208, 214, 229, 230, 231, 258, 302, 332, 335, 338, 356, 396, 398, 399, 405, 434, 465, 485, 488, 517, 520], "content": [9, 396, 517], "context": [309, 519], "contigu": [0, 1, 2, 87, 154, 156], "continu": [362, 444, 518, 520], "contract": [0, 141], "contribut": 2, "contriv": [520, 527], "control": [0, 384, 514, 523], "conv": 114, "conv1d": [0, 356], "conv2d": [0, 356], "conv3d": [0, 356], "conv_gener": 0, "conv_transpose1d": 0, "conv_transpose2d": 0, "conv_transpose3d": 0, "conveni": [1, 2, 7, 198], "convent": [19, 114, 140, 141, 435], "convers": 8, "convert": [0, 1, 2, 82, 89, 90, 91, 125, 175, 260, 414, 415, 523, 524, 525], "convolut": [0, 107, 108, 109, 110, 111, 112, 113, 114, 363, 364, 365, 366, 367, 368, 370, 371, 499], "convolv": [107, 108, 109, 110, 111, 112, 113], "convtranspose1d": 356, "convtranspose2d": 356, "convtranspose3d": 356, "cooperative_group": 154, "coordin": [0, 237], "copi": [0, 1, 2, 6, 8, 106, 254, 301, 521, 524], "core": [1, 2, 3, 4, 5, 6, 7, 336, 356, 358, 359, 360, 361, 381, 388, 389, 390, 399, 402, 404, 407, 414, 415, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 465, 485, 488, 490, 517, 518, 524, 525], "corner": 435, "correct": [2, 9, 494, 495, 496, 521, 523], "correctli": [39, 518], "correl": [110, 370], "correspond": [0, 1, 2, 16, 18, 82, 101, 126, 128, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 207, 208, 232, 234, 242, 256, 298, 299, 311, 318, 326, 331, 340, 498, 518, 520, 522], "cos_first": 426, "cosh": [0, 464], "cosin": [0, 20, 21, 115, 116, 457, 506, 508, 520], "cosine_decai": [490, 508], "cosine_similarity_loss": 356, "cost": [9, 492, 518, 523], "costli": 523, "cot": 1, "cot_index": 1, "cotan": 122, "cotang": [1, 2, 122, 330], "could": [6, 356, 521], "count": [356, 508], "counter": 514, "cours": 520, "coursera": 504, "cout": [4, 519], "cov": 266, "covari": [266, 361], "cover": 2, "cpp": [2, 4], "cpu": [8, 207, 208, 209, 210, 216, 347, 527], "cpython": 2, "crash": [87, 517], "creat": [0, 1, 2, 6, 9, 87, 133, 153, 188, 309, 356, 488, 490, 508, 517, 518, 519, 521, 522, 524], "create_additive_causal_mask": 6, "cross": [7, 110, 456, 458], "cross_entropi": [7, 356], "crowd": 6, "cry": 6, "cubic": 435, "cuda": [8, 154], "cummax": 0, "cummin": 0, "cumprod": 0, "cumsum": 0, "cumul": [0, 87, 118, 119, 120, 121, 228], "current": [6, 8, 9, 87, 96, 109, 112, 113, 126, 137, 182, 238, 258, 342, 356, 492, 518, 523], "current_binary_dir": 2, "custom": [8, 122, 154, 156, 434], "custom_decod": 434, "custom_encod": 434, "custom_funct": 1, "custom_kernel_myexp_float": 1, "custom_tim": 2, "cvpr": 370, "cxx": 4, "cycl": 516, "d": [0, 1, 2, 6, 109, 113, 127, 128, 158, 159, 191, 214, 233, 237, 252, 314, 321, 323, 324, 325, 343, 365, 368, 371, 376, 382, 417, 491, 494, 496, 527], "d1": 527, "d2": 527, "d2fdx2": 520, "d_i": 385, "dampen": 505, "darwin": 2, "data": [0, 2, 7, 8, 11, 19, 135, 153, 171, 172, 178, 188, 221, 246, 250, 263, 272, 321, 323, 329, 333, 371, 436, 437, 438, 439, 440, 441, 442, 443, 517, 518, 519, 521, 524], "dataset": [5, 518, 523], "datatyp": 53, "dbuild_shared_lib": 9, "dcmake_build_typ": [4, 9], "ddof": [0, 79, 84, 307, 328], "deal": 517, "deb": 9, "debug": [1, 3, 518, 522], "debugg": 8, "decai": [492, 495, 497, 499, 505, 506, 507, 510], "decay_r": [492, 507, 510], "decay_step": 506, "decent": 7, "decid": [340, 396], "decim": [0, 70, 282], "declar": 2, "decltyp": 1, "decod": 434, "decomposit": [204, 205, 215, 219], "decor": [1, 122], "decoupl": 495, "dedic": 518, "deep": [361, 437, 438, 439, 440], "def": [1, 2, 5, 6, 7, 122, 150, 152, 154, 156, 327, 356, 488, 517, 518, 519, 520, 521, 523, 524, 527], "default": [1, 2, 9, 15, 16, 17, 18, 19, 27, 28, 29, 30, 86, 87, 96, 102, 103, 107, 108, 109, 110, 111, 112, 113, 122, 123, 124, 126, 127, 128, 131, 132, 133, 135, 136, 137, 150, 152, 153, 154, 156, 158, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 175, 179, 180, 184, 187, 188, 192, 200, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 232, 234, 236, 237, 242, 246, 250, 253, 254, 256, 258, 259, 261, 262, 263, 265, 266, 267, 268, 269, 271, 272, 273, 277, 279, 282, 289, 290, 291, 292, 293, 301, 302, 305, 306, 307, 309, 311, 313, 318, 320, 321, 322, 323, 324, 325, 326, 327, 328, 331, 333, 335, 336, 339, 347, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 372, 375, 376, 378, 381, 382, 384, 385, 388, 389, 390, 392, 397, 399, 404, 406, 409, 410, 411, 412, 413, 414, 415, 417, 421, 426, 430, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 488, 491, 492, 493, 494, 495, 496, 497, 499, 504, 505, 506, 514, 516, 517, 518, 519, 520, 522, 524, 526, 528], "default_devic": 528, "default_stream": 528, "defin": [1, 2, 5, 6, 7, 9, 122, 136, 154, 156, 180, 206, 214, 259, 336, 339, 522, 524], "definit": [122, 204, 205, 266], "degre": [0, 260, 469], "deleg": 498, "delta": [461, 491], "delv": [439, 440], "demonstr": 524, "denomin": [381, 457, 491, 493, 494, 495, 496, 504], "dens": [237, 527], "depend": [0, 2, 3, 4, 5, 9, 82, 214, 376, 382, 417, 517, 518, 521, 526, 527], "depth": [339, 360, 365, 368, 371, 390, 520], "dequant": [0, 258], "deriv": [2, 519, 520, 523], "descend": 394, "descent": [505, 517, 523], "describ": [2, 258, 523], "descript": [2, 4, 6, 347], "design": [1, 5, 8, 514, 527], "destin": [0, 2, 63, 137, 244, 257, 339, 490, 519], "destroi": 517, "detach": 520, "detail": [1, 2, 11, 258, 289, 356, 370, 421, 426, 435, 437, 438, 439, 440, 491, 493, 494, 496, 497, 518, 521, 525], "detect": 517, "determin": [0, 2, 128, 266, 342, 347, 403, 526], "dev": [2, 9], "develop": [2, 4, 9], "developer_dir": 9, "deviat": [0, 267, 307, 437, 439, 442], "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, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 123, 124, 125, 126, 127, 128, 131, 132, 135, 136, 137, 138, 139, 140, 143, 144, 145, 147, 148, 149, 153, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 177, 178, 179, 180, 185, 186, 187, 188, 189, 191, 192, 193, 194, 195, 196, 197, 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, 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, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 282, 283, 290, 291, 292, 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, 328, 329, 332, 333, 334, 344, 527, 528], "device_info": 293, "devicetyp": 10, "df": 524, "dfdx": [519, 520, 521], "dft": [160, 161, 162, 164, 165, 166, 171, 172, 173], "dhwc": 371, "diag": [0, 219], "diagon": [0, 46, 127, 153, 321, 323, 324, 325], "dict": [88, 102, 146, 151, 222, 238, 285, 286, 287, 336, 338, 339, 343, 402, 407, 410, 411, 488, 500, 501, 503, 516, 520, 526], "dict_kei": [340, 501], "dictionari": [6, 102, 150, 190, 222, 238, 285, 286, 338, 339, 342, 356, 396, 405, 410, 411, 502, 516, 526], "did": 6, "diff": 2, "differ": [8, 167, 198, 207, 209, 310, 329, 468, 498, 499, 517, 518, 519, 520, 522], "differenti": [1, 8, 362, 444], "difficult": 520, "difficulti": [437, 438], "dilat": [0, 107, 108, 109, 110, 111, 112, 113, 363, 364, 365, 366, 367, 368], "dim": [1, 6, 158, 159, 373, 377, 381, 383, 412, 414, 416, 421, 426, 434], "dimens": [0, 1, 2, 6, 16, 18, 27, 28, 65, 72, 82, 89, 90, 91, 102, 108, 109, 110, 112, 113, 128, 148, 158, 159, 169, 170, 172, 173, 175, 179, 180, 191, 204, 205, 207, 208, 209, 210, 211, 212, 214, 215, 216, 219, 220, 232, 233, 234, 236, 242, 256, 257, 258, 262, 271, 307, 311, 315, 318, 322, 328, 361, 363, 364, 365, 366, 367, 368, 370, 371, 375, 376, 377, 381, 382, 383, 412, 416, 417, 421, 434, 435, 449, 458, 499, 517, 520], "dimension": [31, 155, 157, 160, 161, 162, 164, 165, 166, 171, 172, 173, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 373, 385, 388, 389, 390, 414, 415, 426, 521, 524], "dir": 4, "direct": [6, 394, 497, 527], "directli": [2, 6, 87], "directori": [2, 4, 6, 9], "disabl": [129, 289, 335, 517, 518], "disable_compil": 517, "disappoint": 6, "discard": [6, 339], "discov": [9, 518], "discoveri": 497, "discret": [114, 160, 161, 162, 164, 165, 166, 171, 172, 173, 373, 414], "discuss": 2, "disk": 6, "dispatch": 2, "dispatch_thread": 2, "dispatchthread": 1, "displai": 356, "distanc": [6, 469], "distribut": [8, 9, 261, 262, 263, 265, 266, 267, 272, 273, 335, 385, 437, 438, 439, 440, 442, 443, 459, 462, 467, 469, 485], "distributed_config": [518, 522], "diverg": 462, "divid": [0, 2, 39, 177, 258, 276, 518], "divis": [0, 138, 177, 258, 276], "divisor": [307, 328], "divmod": 0, "dloss_dw": 520, "dloss_dx": 520, "dlpack": 524, "dlvalu": 327, "dmlx_build_cpu": 9, "dmlx_build_cuda": 9, "dmlx_build_gguf": 9, "dmlx_build_safetensor": 9, "dmlx_metal_debug": 3, "dmlx_metal_jit": 9, "do": [0, 2, 6, 9, 212, 329, 356, 398, 409, 485, 488, 517, 518, 519, 520, 523], "doc": [2, 7, 518, 522], "document": [2, 3, 4, 69, 83, 156, 285, 286, 347, 517, 518, 519, 520, 521], "doe": [0, 2, 3, 6, 9, 181, 258, 329, 338, 356, 517, 518, 521, 522, 523, 524], "doesn": [2, 356, 519], "domain": 272, "don": [1, 9, 106, 517, 527], "done": [356, 369, 416, 517, 518, 523, 524], "dot": [151, 211, 220, 318, 339, 401, 412, 518], "doubl": [0, 6, 347], "doubt": 6, "down": [6, 338], "download": 9, "downsampl": [358, 359, 360, 388, 389, 390], "dparam": 327, "dpkg": 9, "draw": 262, "driver": 9, "drop": 396, "dropout": [356, 370, 371, 406, 434, 517], "dropout2d": 356, "dropout3d": 356, "dst": 137, "dt": 144, "dtype": [0, 1, 2, 6, 12, 19, 31, 38, 39, 82, 85, 135, 136, 153, 154, 156, 174, 175, 178, 188, 198, 200, 207, 208, 209, 210, 214, 216, 221, 250, 263, 265, 266, 267, 269, 272, 273, 298, 299, 321, 323, 326, 329, 333, 335, 347, 404, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 465, 506, 507, 508, 509, 510, 517, 518, 519, 520, 521, 524, 525, 526], "dtypecategori": [198, 347], "dual": 464, "duchi": 493, "duplic": 519, "dure": [3, 102, 292, 369, 370, 371, 435, 524], "dx": 122, "dy": 122, "dyld": 518, "dyld_library_path": 518, "dylib": 2, "dynam": [0, 154, 519, 523], "e": [2, 7, 9, 122, 144, 156, 179, 180, 199, 294, 361, 363, 364, 365, 366, 367, 368, 370, 371, 377, 381, 383, 397, 416, 454, 455, 478, 479, 484, 490, 493, 499, 517, 519, 523, 528], "e2m1": 258, "e5": 347, "e8": 347, "each": [0, 1, 2, 72, 88, 126, 146, 158, 180, 198, 204, 205, 207, 208, 209, 210, 211, 215, 219, 220, 233, 237, 253, 258, 259, 262, 277, 287, 288, 302, 319, 322, 329, 331, 332, 370, 371, 373, 376, 377, 382, 417, 421, 434, 456, 458, 514, 517, 518, 519, 522, 523], "eager": 523, "earli": 370, "eas": 6, "easi": [2, 356, 498], "easier": [1, 151, 523], "easiest": 518, "edg": [101, 253, 435, 517], "edit": [9, 411], "effect": [370, 517, 523], "effici": [6, 8, 179, 370, 421, 518, 523, 525], "eigenvalu": [207, 208, 209, 210], "eigenvector": [207, 208], "einstein": [140, 141], "einsum": 141, "either": [9, 14, 69, 82, 83, 92, 94, 95, 101, 138, 139, 143, 150, 177, 185, 186, 190, 201, 202, 203, 214, 227, 233, 235, 243, 245, 249, 255, 258, 276, 280, 310, 327, 359, 360, 389, 390, 423, 435, 439, 440, 518, 522, 524], "elem": [1, 154, 156], "elem_to_loc": [1, 2], "element": [0, 1, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 29, 74, 87, 92, 93, 94, 95, 99, 115, 116, 118, 119, 120, 121, 126, 138, 139, 143, 144, 145, 147, 149, 153, 176, 177, 180, 185, 186, 192, 193, 194, 195, 196, 197, 201, 202, 203, 223, 224, 225, 226, 227, 228, 229, 230, 231, 235, 237, 243, 245, 247, 249, 254, 255, 258, 259, 275, 276, 277, 280, 281, 283, 294, 295, 296, 297, 303, 304, 310, 314, 316, 317, 320, 327, 329, 332, 362, 369, 370, 371, 376, 380, 382, 391, 413, 417, 421, 425, 444, 451, 452, 454, 455, 470, 471, 473, 474, 477, 478, 479, 480, 517, 520, 521], "elementwis": [1, 104, 105], "elif": 6, "ellipsi": 521, "elman": 417, "els": [0, 2, 6, 356, 397, 518, 523], "elsewher": [323, 521], "elu": [356, 475], "emb": [6, 373, 414, 426], "embed": [6, 336, 356, 414, 421, 426, 457, 499], "empti": [266, 339], "en0": 522, "en2": 518, "enabl": [3, 6, 9, 102, 142, 335, 499, 505], "enclos": 519, "encod": [2, 158, 421, 426, 434, 458], "encount": [2, 520], "end": [117, 128, 205, 239, 258, 281, 335, 376, 382, 432, 450, 461, 468, 475, 481, 482, 506, 509, 519], "end_axi": [0, 50, 175], "end_encod": 2, "endif": 2, "endl": [4, 519], "endswith": 397, "enhanc": [6, 421, 523], "enough": [2, 523], "ensur": [0, 1, 2, 9, 154, 156, 338, 464, 518, 519], "ensure_row_contigu": [1, 154, 156], "enter": 6, "entir": [16, 18, 27, 28, 232, 234, 236, 242, 256, 307, 311, 328, 370, 371], "entri": [0, 268, 326, 370, 371], "entropi": [7, 456, 458], "enumer": 356, "environ": [9, 129, 142, 518], "ep": [5, 155, 157, 361, 377, 381, 383, 416, 457, 459, 469, 490, 491, 492, 493, 494, 495, 496, 504], "epoch": 7, "epsilon": [361, 377, 381, 383, 416, 457, 459, 491, 493, 494, 495, 496, 504], "epsilon_1": 492, "epsilon_2": 492, "equal": [0, 1, 17, 29, 86, 153, 186, 192, 203, 249, 254, 269, 302, 335, 381, 385], "equal_nan": [0, 17, 86, 192], "equat": [140, 141, 217, 218], "equival": [0, 2, 32, 69, 83, 136, 139, 154, 177, 180, 187, 314, 362, 372, 374, 378, 379, 380, 386, 387, 411, 413, 415, 418, 419, 420, 422, 424, 427, 428, 429, 430, 431, 433, 518], "erf": [0, 145, 517], "erfinv": 0, "error": [0, 2, 9, 133, 144, 145, 293, 302, 374, 446, 447, 448, 464, 466, 517, 520, 522, 524], "error_norm": 5, "estim": 496, "eta": 497, "etc": [2, 258, 356, 435, 518], "ethernet": [518, 522], "eval": [2, 3, 5, 6, 7, 356, 488, 490, 517, 518, 519, 520, 523, 525], "eval_cpu": 2, "eval_fn": 7, "eval_gpu": 2, "evalu": [2, 6, 7, 8, 88, 137, 146, 199, 292, 330, 356, 395, 406, 488, 490, 517, 519, 525], "even": [1, 2, 6, 102, 167, 207, 209, 517, 518, 519, 523, 524], "evenli": [0, 221], "everi": [1, 258, 340, 490, 510, 520, 522], "everyth": [6, 518], "everywher": 0, "exact": [447, 448], "exactli": [2, 6, 158, 399, 520], "exampl": [0, 3, 4, 5, 6, 7, 9, 19, 39, 88, 122, 133, 150, 151, 152, 154, 156, 158, 159, 175, 190, 198, 200, 207, 208, 209, 210, 214, 216, 298, 299, 309, 314, 326, 338, 341, 342, 356, 358, 359, 360, 361, 381, 388, 389, 390, 397, 399, 406, 409, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 465, 485, 490, 501, 506, 507, 508, 509, 510, 514, 520, 521, 522, 523, 524, 525, 526], "exce": [335, 338], "exceed": 292, "except": [8, 122, 153, 168, 169, 171, 172, 173, 292, 347, 377, 399, 519, 521, 524], "exclud": [257, 315], "exclus": [0, 87, 95], "execut": [2, 4, 9, 89, 90, 91, 183, 200, 518, 524, 527], "execute_process": 4, "exist": [2, 3, 6, 397, 409, 518], "exp": [0, 1, 88, 149, 154, 156, 227, 232, 263, 300, 362, 372, 425, 444, 445, 462, 475, 476, 480, 517, 519, 527], "exp_elementwis": [1, 154, 156], "expand_dim": 0, "expect": [6, 158, 363, 364, 365, 366, 367, 368, 369, 370, 371, 426, 434, 459, 517, 518, 521], "expens": 434, "expensive_fun": 523, "experiment": [88, 150, 152, 190, 524], "explain": 2, "explicit": [2, 501, 514, 524], "explicitli": [179, 356, 514, 522], "explor": 9, "expm1": 0, "exponenti": [0, 147, 149, 362, 372, 422, 444, 445, 475, 507], "exponential_decai": 490, "export": [8, 9, 150, 151, 190], "export_funct": 519, "ext_modul": 2, "extend": [2, 253], "extens": [8, 222, 240, 403, 526], "extern": 524, "extra": [1, 340, 341, 519], "extract": [0, 6, 46, 127, 128, 298, 356, 396, 488], "extras_requir": 2, "extrem": [521, 523], "ey": [0, 6, 211, 220], "f": [0, 2, 5, 7, 122, 214, 356, 382, 495, 517, 524], "f_jvp": 122, "f_t": 382, "f_vjp": 122, "f_vmap": 122, "face": 6, "factor": [2, 15, 187, 204, 205, 212, 213, 216, 435, 458, 507, 510], "fail": [517, 518, 522], "fall": [2, 122], "fallback": 498, "fals": [0, 1, 2, 6, 16, 17, 18, 27, 28, 34, 35, 36, 37, 42, 43, 44, 45, 58, 59, 60, 61, 62, 66, 79, 80, 84, 86, 102, 106, 110, 118, 119, 120, 121, 133, 150, 152, 156, 179, 180, 192, 198, 204, 205, 214, 218, 219, 220, 222, 228, 232, 234, 236, 237, 242, 256, 307, 311, 328, 332, 336, 339, 340, 341, 342, 347, 377, 381, 383, 385, 397, 399, 409, 412, 415, 421, 426, 434, 435, 456, 459, 492, 494, 495, 505, 519, 524], "famili": 6, "fan": [437, 438, 439, 440], "fan_in": [437, 438, 439, 440], "fan_out": [437, 438, 439, 440], "far": 490, "fast": [1, 8, 374, 448, 518, 527], "faster": [1, 2, 9, 139, 179, 180, 446, 456, 517, 518, 520], "featur": [1, 8, 107, 108, 109, 110, 111, 112, 113, 158, 361, 376, 377, 381, 382, 383, 385, 415, 416, 417, 421, 434, 435, 517, 518, 523], "feed": 6, "feed_forward": 6, "feedforward": [437, 438], "feel": 6, "fetch": 1, "few": [1, 2, 6, 7, 8, 9, 519, 523, 525], "fewer": 518, "ffn": 6, "ffn_norm": 6, "fft": 8, "fftshift": 167, "fi": 518, "figur": 518, "file": [4, 6, 9, 150, 151, 152, 190, 222, 284, 285, 286, 287, 288, 399, 403, 518, 519, 520, 526], "file_or_weight": 399, "fill": [0, 178, 251, 323, 334, 436, 437, 438, 439, 440, 442, 443], "filter": [0, 114, 363, 364, 365, 366, 367, 368, 392, 396, 498, 499], "filter_and_map": 356, "filter_fn": [392, 396], "final": [2, 4, 5, 6, 7, 187, 499, 506, 509, 518, 522], "find": [2, 4, 5, 9, 518], "find_packag": [2, 4], "finder": 9, "fine": [514, 519, 523], "finetun": 356, "finit": [0, 193, 246], "first": [0, 1, 2, 3, 4, 5, 6, 7, 9, 128, 131, 133, 175, 184, 198, 200, 201, 219, 229, 231, 233, 254, 271, 280, 312, 318, 321, 327, 339, 341, 342, 356, 359, 360, 377, 389, 390, 435, 457, 465, 492, 496, 501, 517, 518, 519, 520, 521, 522, 524, 527], "first_lay": 523, "firt": 517, "fit": [258, 527], "five": 517, "fix": [2, 6, 9, 517, 523], "flag": [4, 9, 179, 517, 524], "flat": [179, 180, 339, 343], "flat_param": 287, "flatten": [0, 29, 30, 118, 119, 120, 121, 214, 228, 252, 254, 257, 277, 281, 301, 314, 315, 320, 339, 499, 517], "flexibl": 8, "flexibli": 411, "flip": [0, 110, 114], "float": [0, 1, 2, 12, 15, 17, 19, 82, 155, 156, 157, 158, 159, 174, 177, 178, 187, 192, 198, 214, 246, 258, 259, 261, 265, 335, 338, 347, 361, 369, 370, 371, 377, 381, 383, 392, 404, 416, 421, 426, 432, 434, 435, 436, 437, 438, 439, 440, 442, 443, 457, 458, 459, 461, 465, 468, 469, 481, 482, 491, 492, 493, 494, 495, 496, 497, 499, 504, 505, 506, 507, 509, 510], "float16": [1, 2, 12, 154, 156, 187, 222, 347, 392, 523, 524], "float16_t": [1, 2], "float32": [0, 1, 2, 12, 19, 153, 154, 156, 159, 187, 188, 198, 208, 210, 214, 216, 221, 250, 263, 265, 266, 267, 272, 273, 299, 323, 333, 347, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 458, 465, 506, 507, 508, 509, 510, 517, 518, 519, 520, 521, 523, 524, 525, 526], "float64": [12, 198, 347, 524], "floor": [0, 1, 177], "floor_divid": 0, "flow": [0, 308, 523], "flush": 2, "fn": [190, 337, 340, 341, 342, 525], "follow": [1, 2, 4, 6, 7, 8, 9, 19, 114, 126, 159, 179, 214, 253, 258, 341, 356, 447, 448, 462, 491, 492, 493, 496, 497, 499, 505, 514, 517, 518, 519, 520, 522, 527], "food": 6, "forc": [6, 7, 106, 356, 518, 525], "forg": 518, "format": [6, 151, 222, 258, 284, 285, 286, 287, 288, 518, 524], "formul": [362, 372], "formula": 468, "forth": [435, 518], "forward": [1, 2, 327, 517, 522, 523], "found": [4, 258, 396], "four": 361, "fourier": [160, 161, 162, 164, 165, 166, 171, 172, 173], "fourth": 519, "frac": [144, 258, 267, 294, 361, 369, 370, 371, 377, 381, 383, 385, 416, 425, 437, 438, 439, 440, 457, 459, 461, 464, 476, 478, 479, 491, 493, 494, 495, 496, 504], "fraction": 19, "framework": 8, "free": 289, "freez": [356, 409, 488], "freq": 158, "frequenc": [158, 163, 421, 426], "frequent": [517, 523], "friend": 6, "fro": 214, "frobeniu": 214, "from": [0, 1, 2, 4, 6, 7, 8, 87, 125, 126, 128, 131, 132, 135, 136, 137, 154, 156, 169, 170, 172, 173, 178, 179, 183, 187, 190, 207, 209, 214, 222, 233, 237, 251, 258, 260, 261, 262, 263, 264, 265, 269, 272, 287, 289, 298, 305, 308, 310, 314, 315, 320, 321, 332, 334, 339, 340, 341, 342, 343, 356, 385, 397, 399, 412, 437, 438, 439, 440, 442, 443, 459, 468, 485, 490, 516, 517, 518, 519, 520, 523, 524, 525, 526, 527], "from_embed": 414, "from_linear": 415, "front": [2, 519], "frozen": [356, 397, 407, 409, 415, 488], "fuction": 139, "full": [0, 1, 2, 7, 69, 83, 114, 156, 212, 300, 410, 411, 459, 498, 517, 518, 519, 523], "full_turn": 426, "fulli": [2, 8, 499, 522, 524, 527], "fun": [102, 150, 152, 184, 199, 327, 330, 331, 517, 519, 521, 523, 527], "fun1": 523, "func": 417, "function": [0, 1, 2, 3, 5, 6, 7, 8, 17, 19, 87, 102, 122, 139, 144, 145, 150, 152, 154, 156, 184, 190, 192, 199, 204, 205, 207, 208, 209, 210, 211, 214, 215, 216, 219, 220, 233, 293, 294, 327, 330, 331, 337, 338, 340, 341, 342, 356, 362, 372, 374, 375, 378, 379, 380, 386, 387, 391, 393, 397, 404, 409, 413, 417, 418, 419, 420, 422, 423, 424, 425, 427, 428, 429, 430, 431, 432, 433, 434, 446, 447, 448, 449, 450, 451, 452, 454, 455, 456, 470, 473, 476, 478, 479, 480, 481, 482, 483, 485, 490, 501, 514, 516, 518, 521, 523, 524, 526], "functionexport": 152, "functool": 517, "further": [2, 9, 520], "fuse": [1, 517], "fusibl": 517, "futur": [6, 88, 150, 152, 190, 415, 521, 523], "fx": 122, "g": [3, 9, 122, 156, 214, 258, 382, 484, 499, 504, 505, 519, 523, 528], "g_t": [382, 491, 493, 494, 495, 496, 497, 504, 505], "gain": [437, 438, 439, 440], "gamma": [361, 377, 381, 383, 416, 437, 438, 439, 440], "gap": 1, "gate": [375, 376, 449], "gather": [0, 131, 179, 180], "gather_mm": [0, 180], "gather_qmm": 0, "gaurante": 329, "gaussian": [5, 374, 446, 447, 448, 459], "gaussian_nll_loss": 356, "gc_func": 434, "gelu": [356, 447, 448, 517], "gelu_approx": [356, 374, 446], "gelu_fast_approx": [356, 374, 446], "geluapprox": 374, "gelufast": 374, "gener": [0, 1, 3, 5, 12, 19, 110, 153, 154, 156, 169, 170, 215, 221, 237, 261, 266, 267, 268, 269, 272, 273, 434, 514, 517, 521, 523, 528], "generate_stub": 9, "geq": [432, 482], "get": [2, 5, 7, 9, 108, 109, 110, 112, 113, 123, 124, 174, 181, 182, 183, 238, 264, 356, 517, 519, 520, 523, 527], "get_cache_memori": 100, "get_command_encod": 2, "get_kernel": 2, "get_librari": 2, "gguf": [9, 222, 285, 526], "gh": 1, "gii": 1, "git": 9, "github": [5, 7, 9, 517], "give": [2, 6, 7, 29, 517], "given": [0, 2, 9, 16, 18, 29, 39, 87, 88, 98, 101, 103, 118, 119, 120, 121, 126, 128, 141, 146, 148, 158, 159, 160, 161, 162, 164, 165, 166, 171, 172, 173, 178, 179, 212, 214, 228, 232, 234, 236, 242, 246, 248, 256, 266, 268, 269, 281, 282, 289, 291, 300, 302, 307, 311, 313, 319, 320, 321, 323, 324, 325, 328, 344, 369, 396, 412, 457, 459, 465, 498], "gix": 1, "gix_mult": 1, "giy_mult": 1, "glibc": 9, "global": [129, 131, 132, 133, 135, 136, 137, 142, 270, 335, 338, 514, 517], "glorot": [437, 438], "glorot_norm": 356, "glorot_uniform": 356, "glu": [6, 356], "gm": 1, "gn": 1, "go": [2, 6, 518, 520], "golub": 214, "good": [2, 9, 490, 517, 518, 522, 527], "goroshin": 370, "gower": 6, "gpu": [1, 3, 8, 9, 238, 347, 521, 527], "gputrac": [3, 240], "grad": [2, 5, 7, 122, 327, 338, 490, 500, 517, 518, 519, 520, 521, 523, 525], "grad_fn": [5, 517, 520], "gradient": [0, 5, 7, 122, 184, 308, 327, 335, 337, 338, 356, 397, 410, 415, 434, 464, 488, 490, 491, 492, 494, 495, 496, 497, 500, 503, 505, 517, 518, 520, 521, 523, 524, 525], "grain": 514, "graph": [2, 6, 7, 8, 151, 292, 519, 520], "great": 3, "greater": [0, 6, 29, 149, 186, 254, 338, 432, 482], "greater_equ": 0, "grep": 9, "grid": [2, 154, 156, 237], "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, 458, 468], "group": [0, 1, 107, 108, 109, 110, 111, 112, 113, 126, 131, 132, 133, 135, 136, 137, 159, 180, 258, 259, 329, 335, 336, 363, 364, 377, 414, 415, 518], "group_dim": 2, "group_siz": [0, 126, 180, 258, 259, 336, 414, 415], "groupnorm": 356, "grow": 523, "gru": 356, "guid": [2, 4, 8, 518, 519], "guidelin": 292, "gw": 1, "h": [1, 2, 4, 108, 109, 112, 113, 214, 361, 364, 365, 367, 368, 370, 371, 376, 382, 417, 520, 523], "h_": [376, 382, 417], "h_in": 1, "h_stride": 1, "h_t": [376, 382, 417], "ha": [2, 3, 6, 7, 8, 9, 82, 102, 128, 137, 168, 169, 171, 172, 173, 182, 184, 204, 205, 207, 208, 209, 210, 211, 215, 219, 220, 237, 262, 361, 376, 382, 385, 417, 488, 490, 517, 518, 519, 521, 522, 523, 525, 527], "had": 6, "hadamard": [0, 187], "hadamard_transform": 0, "half": [19, 269, 273, 421, 523], "halv": [375, 449], "hand": [6, 520, 523], "handi": 520, "handl": [2, 356, 517], "happen": [2, 6, 155, 157, 434, 490, 517, 523], "happi": 6, "hard": 6, "hard_shrink": [356, 378], "hard_tanh": [356, 379], "hardcod": 517, "hardshrink": [356, 450], "hardswish": 356, "hardtanh": [356, 451], "hat": [126, 258], "have": [0, 1, 2, 6, 9, 17, 86, 89, 90, 91, 96, 122, 131, 159, 169, 170, 172, 173, 180, 192, 233, 240, 258, 262, 329, 335, 339, 382, 412, 423, 497, 501, 516, 517, 518, 519, 521, 522, 523, 527], "haven": 6, "hazan": 493, "he": [6, 439, 440], "he_norm": 356, "he_uniform": 356, "head": [159, 412, 434], "header": [2, 9, 154, 156], "heart": 6, "heavi": 6, "height": [359, 360, 361, 364, 365, 367, 368, 370, 371, 389, 390], "hello": [339, 343], "help": [2, 6, 517, 527], "helper": [6, 154, 156, 335, 517, 518, 522], "henc": [0, 2, 258, 517], "hendryck": 448, "here": [2, 6, 9, 490, 517, 519, 520, 523, 526, 527], "hermitian": [208, 210], "hf": 382, "hg": 382, "hh": 417, "hi": [6, 382], "hidden": [376, 382, 417, 434, 499], "hidden_dim": [7, 488, 490], "hidden_s": [376, 382, 417], "hierarchi": 347, "high": [269, 273, 356, 373, 443, 485], "high_pad_s": 0, "higher": [2, 191, 293, 465, 518, 520], "highli": 9, "him": 6, "hing": 460, "hinge_loss": 356, "hinton": 504, "hit": 2, "hn": 376, "ho": 382, "hold": [2, 6, 11, 12, 214, 517], "homebrew": 518, "hopkin": 214, "host": 2, "host1": 518, "host2": 518, "host3": 518, "host4": 518, "host_nam": 1, "hostfil": [518, 522], "hostnam": [518, 522], "hostname1": [518, 522], "hostname2": [518, 522], "hostname3": 518, "hostname4": 518, "hot": 458, "hour": 6, "how": [2, 4, 6, 7, 356, 358, 359, 360, 363, 364, 365, 366, 367, 368, 373, 388, 389, 390, 414, 435, 500, 517, 521, 527], "howev": [2, 122, 356, 374, 377, 501, 514, 517, 518, 523, 524], "hr": 376, "http": [9, 377, 381, 383, 391, 416, 448, 470], "huber": 461, "huber_loss": 356, "human": [439, 440], "hundr": 9, "hurri": 6, "hutter": 495, "hyperbol": [0, 21, 23, 26, 116, 297, 317, 433, 483], "hz": 376, "i": [0, 1, 2, 3, 4, 6, 7, 8, 9, 17, 19, 29, 38, 82, 87, 88, 101, 108, 109, 110, 112, 113, 114, 117, 118, 119, 120, 121, 122, 127, 128, 131, 132, 134, 135, 136, 137, 139, 146, 150, 152, 154, 155, 156, 157, 158, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 175, 177, 178, 179, 180, 187, 190, 192, 193, 198, 199, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 222, 227, 228, 232, 233, 237, 239, 253, 254, 257, 258, 259, 266, 267, 268, 279, 281, 284, 285, 286, 292, 293, 294, 300, 302, 307, 308, 313, 314, 315, 318, 321, 322, 326, 327, 328, 329, 330, 331, 332, 335, 336, 338, 339, 340, 341, 342, 347, 349, 356, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 369, 370, 371, 374, 376, 377, 381, 382, 383, 385, 388, 389, 390, 396, 397, 403, 405, 406, 408, 409, 410, 411, 412, 413, 415, 416, 417, 421, 426, 432, 434, 435, 439, 440, 446, 448, 456, 457, 459, 464, 465, 468, 469, 471, 477, 482, 488, 490, 492, 494, 495, 497, 498, 500, 501, 506, 508, 509, 514, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528], "i386": 9, "i_n": 1, "i_nw": 1, "i_s": 1, "i_sw": 1, "i_t": 382, "iclr": [494, 495, 496], "id": [7, 9], "idea": [520, 523], "idempot": [397, 409], "ident": [0, 122, 137, 153, 167, 308, 356, 406, 518], "identifi": [2, 339, 516], "idim": 7, "idiom": [7, 517], "idx": [39, 521], "ie": 409, "ieee": 347, "ifac": 518, "ignor": [6, 39, 88, 101, 102, 146, 492, 522], "ih": 417, "ii": 1, "ij": 237, "im": 267, "imag": [0, 364, 365, 367, 368, 370, 371, 435], "imagenet": [439, 440], "imaginari": [51, 189], "immedi": [6, 392], "implement": [0, 1, 5, 7, 158, 159, 179, 180, 373, 396, 412, 421, 423, 426, 432, 434, 435, 482, 491, 492, 493, 496, 497, 499, 500, 512, 517, 520], "impli": 329, "implicit": [514, 517, 520], "implicitli": 523, "import": [2, 3, 5, 6, 7, 9, 122, 133, 187, 190, 214, 287, 327, 339, 340, 341, 342, 343, 356, 358, 359, 360, 361, 381, 388, 389, 390, 399, 435, 456, 458, 465, 485, 488, 490, 517, 518, 520, 521, 523, 524, 525], "import_funct": 519, "imported_ab": 519, "imported_fun": 519, "imported_funct": 519, "improv": [1, 2, 3, 6, 456, 491, 492, 493, 494, 495, 496, 504, 517, 518], "in_ax": [331, 520], "in_channel": [363, 364, 365, 366, 367, 368], "in_dim": [356, 488], "in_proj": 488, "includ": [1, 2, 4, 118, 119, 120, 121, 151, 154, 156, 181, 182, 228, 292, 383, 393, 405, 415, 459, 490, 517, 519, 520, 521, 525, 526, 528], "include_dir": 2, "inclus": [0, 42, 43, 44, 45, 58, 118, 119, 120, 121, 175, 228], "incom": 2, "inconveni": 517, "incorpor": 524, "incorrect": 524, "increas": [293, 522], "increment": 19, "incur": [6, 9], "independ": [130, 370, 371], "index": [0, 1, 2, 8, 10, 29, 39, 148, 153, 179, 184, 237, 254, 298, 299, 314, 315, 327], "indic": [0, 2, 17, 27, 28, 29, 30, 39, 159, 179, 180, 184, 192, 193, 194, 195, 196, 197, 198, 212, 219, 257, 298, 299, 302, 314, 315, 327, 406, 408, 458, 465, 508, 521], "indices_or_sect": [75, 302], "indirectli": 524, "individu": [356, 370, 371], "ineffici": [521, 523], "inexact": [12, 198], "inf": [214, 246, 412], "infer": [8, 178, 222, 321, 326, 518, 519], "infin": [0, 194, 196, 197, 246, 388, 389, 390, 496], "infinit": [17, 192, 193], "info": [6, 9], "inform": [3, 4, 6, 7, 9, 141, 174, 238, 285, 286, 347, 356, 361, 374, 412, 518, 519, 520, 527], "inherit": [7, 516], "inifn": 194, "init": [356, 413, 485, 490, 506, 507, 509, 510, 518], "init_fn": [436, 437, 438, 439, 440, 441, 442, 443, 485], "init_valu": 1, "initi": [1, 3, 5, 6, 133, 342, 356, 361, 377, 381, 383, 385, 413, 416, 436, 437, 438, 439, 440, 441, 442, 443, 488, 501, 506, 507, 509, 510, 517, 518, 519, 523], "initializer_list": 0, "inject": 0, "inlin": 0, "inner": [0, 517], "inorm": 381, "inp": [1, 154, 156], "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, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 122, 125, 127, 128, 131, 132, 137, 138, 139, 140, 141, 143, 144, 145, 147, 148, 149, 150, 151, 152, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 177, 179, 180, 184, 185, 186, 187, 189, 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, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 242, 243, 244, 245, 246, 247, 249, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 268, 271, 274, 275, 276, 277, 279, 280, 281, 282, 283, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 314, 315, 316, 317, 318, 319, 320, 321, 322, 324, 325, 326, 327, 328, 329, 331, 332, 334, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 370, 371, 373, 375, 376, 377, 381, 382, 383, 385, 388, 389, 390, 412, 415, 416, 417, 421, 432, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 459, 460, 461, 462, 464, 465, 467, 469, 482, 485, 517, 519, 520, 521, 522, 525, 526], "input_dil": [0, 110], "input_dim": [7, 356, 385, 415], "input_nam": [1, 154, 156], "input_s": [376, 382, 417], "inputs1": 465, "inputs2": 465, "insert": [128, 148, 527], "insid": [517, 519], "inspect": [3, 517, 525], "inspir": 8, "instabl": 469, "instal": [2, 4, 522], "instanc": [6, 39, 122, 258, 343, 356, 381, 392, 393, 394, 397, 399, 400, 401, 406, 409, 410, 411, 423, 488, 518, 522, 524], "instancenorm": 356, "instanti": [1, 2, 7, 523], "instantiate_kernel": 2, "instead": [2, 9, 122, 356, 411, 426, 520, 523], "instruct": [4, 9, 519], "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, 58, 59, 60, 61, 62, 63, 66, 70, 72, 75, 78, 79, 80, 81, 82, 84, 87, 96, 98, 103, 107, 108, 109, 110, 111, 112, 113, 118, 119, 120, 121, 126, 127, 128, 135, 136, 137, 141, 148, 153, 154, 158, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 178, 180, 181, 182, 183, 184, 188, 198, 206, 214, 221, 228, 232, 234, 236, 238, 242, 244, 250, 253, 254, 256, 257, 258, 259, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 277, 279, 281, 282, 289, 292, 293, 298, 299, 300, 301, 302, 305, 306, 307, 311, 312, 314, 315, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 331, 333, 335, 336, 356, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 373, 375, 376, 377, 381, 382, 383, 385, 388, 389, 390, 412, 414, 415, 416, 417, 421, 426, 434, 449, 457, 458, 462, 467, 469, 488, 499, 506, 508, 509, 510], "int16": 347, "int32": [0, 1, 12, 19, 39, 175, 198, 200, 214, 269, 298, 326, 347, 435, 521, 525], "int64": [12, 347], "int64_t": 2, "int8": [12, 347], "int_0": 144, "integ": [0, 12, 177, 179, 180, 198, 214, 238, 253, 258, 259, 261, 268, 269, 302, 314, 318, 331, 347, 373, 404, 508, 521], "integr": [19, 314, 523], "intend": [0, 517], "interact": 434, "interest": 527, "interfac": [2, 518, 522], "intermedi": 524, "intern": 361, "interpol": 435, "interpret": 4, "interv": [19, 221, 269, 273], "introduc": [0, 281], "intuit": 356, "invalid": [0, 87], "invers": [0, 20, 21, 22, 23, 24, 25, 26, 93, 145, 164, 165, 166, 167, 168, 169, 170, 205, 211, 215, 220], "invert": 0, "involv": [490, 517], "iogpu": 293, "iostream": 4, "ip": [518, 522], "ip1": [518, 522], "ip2": [518, 522], "ip3": 518, "ip4": 518, "is_avail": 133, "is_equival": 2, "is_leaf": [339, 340, 341, 342], "is_leaf_fn": 396, "isclos": 0, "isfinit": 0, "ish": 6, "ishmael": 6, "isinf": 0, "isnan": 0, "isneginf": 0, "isposinf": 0, "issu": [518, 520, 524], "issubdtyp": [2, 12, 347], "item": [0, 2, 5, 6, 7, 340, 519, 523, 524, 525], "iter": [5, 7, 219, 340, 341, 499, 514, 517, 523], "iterm": 9, "itertool": [6, 340], "its": [0, 1, 2, 9, 159, 205, 215, 233, 254, 271, 323, 337, 343, 356, 415, 490, 494, 495, 496, 518, 523, 524, 527], "itself": [2, 336, 498, 501], "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, 214, 370, 493, 494, 496], "j8": 2, "jacobian": [2, 199, 330, 525], "jain": 370, "jax": [8, 514], "jit": [1, 154, 156], "jmlr": 493, "jnp": 524, "john": 214, "join": 508, "join_schedul": 490, "jointli": 266, "json": [518, 522], "just": [2, 4, 7, 383, 517, 519, 521], "jvp": [2, 122, 525], "k": [0, 6, 46, 96, 107, 111, 127, 153, 159, 179, 187, 320, 323, 324, 325, 385, 397], "kaim": 440, "kd": [109, 113], "keep": [2, 16, 18, 27, 28, 232, 234, 236, 242, 256, 307, 311, 328, 356, 396, 520, 523], "keepdim": [0, 16, 18, 27, 28, 34, 35, 36, 37, 59, 60, 61, 62, 66, 79, 80, 84, 214, 232, 234, 236, 242, 256, 300, 307, 311, 328], "kei": [1, 3, 6, 150, 159, 190, 238, 261, 262, 263, 265, 266, 267, 268, 269, 271, 272, 273, 339, 340, 396, 397, 409, 412, 501, 514, 516, 519, 520], "kept": 293, "kernel": [2, 8, 9, 107, 108, 109, 110, 111, 112, 113, 154, 156, 358, 388, 517, 521], "kernel_dil": [0, 110], "kernel_s": [358, 359, 360, 363, 364, 365, 366, 367, 368, 388, 389, 390], "key_cach": 6, "key_input_dim": 412, "key_proj": 6, "keyring_1": 9, "keyword": [150, 184, 287, 288, 327, 340, 356, 514, 519, 526, 528], "kh": [108, 109, 112, 113], "kind": 6, "kingma": [494, 496], "kl_div_loss": 356, "kname": 2, "know": [2, 6], "known": [424, 477], "kron": 0, "kroneck": [0, 200], "kth": [0, 29, 254], "kullback": 462, "kw": [108, 109, 112, 113], "kw_onli": 2, "kwarg": [11, 130, 150, 151, 190, 287, 288, 344, 519, 528], "l": [6, 7, 107, 111, 204, 205, 208, 210, 212, 356, 361, 363, 366, 376, 382, 417, 468], "l1": [327, 461, 463, 464, 468], "l1_loss": 356, "l2": [461, 464, 499, 505], "l2_loss": 356, "l_": 461, "la": 214, "label": [3, 5, 458, 465], "label_smooth": 458, "lack": 521, "lambd": [378, 430, 450, 481], "lambda": [340, 341, 342, 356, 378, 392, 397, 404, 430, 450, 475, 481, 491, 492, 493, 494, 495, 496, 497, 504, 505, 517, 518, 519, 520], "languag": [1, 2, 4], "lapack": 9, "larg": [6, 356, 412, 464, 517, 519, 523], "larger": [1, 158, 293, 421, 497], "largest": [214, 246, 320], "lasso": 327, "last": [0, 1, 6, 30, 82, 155, 157, 162, 166, 169, 170, 172, 173, 175, 179, 180, 183, 191, 204, 205, 207, 208, 209, 210, 211, 215, 216, 219, 220, 233, 262, 301, 318, 329, 363, 364, 365, 366, 367, 368, 370, 371, 377, 435, 498, 499, 524], "later": [3, 9, 490], "launch": [1, 2, 133, 518, 521], "layer": [8, 155, 336, 356, 358, 359, 360, 370, 371, 376, 377, 382, 383, 385, 388, 389, 390, 406, 411, 414, 415, 417, 423, 434, 484, 488, 499, 519, 522], "layer_s": 7, "layernorm": 356, "layout": 1, "lazi": [8, 488, 525], "lazili": [6, 356], "lceil": 96, "ld": [376, 382, 417], "lead": [0, 19, 87, 517], "leaf": [102, 336, 339, 340, 341, 342, 396], "leaf_modul": 356, "leaki": [384, 453], "leaky_relu": 356, "leakyrelu": 356, "learn": [5, 7, 8, 361, 377, 381, 383, 413, 416, 490, 491, 492, 493, 494, 495, 496, 497, 499, 504, 505], "learnabl": [363, 364, 365, 366, 367, 368, 423], "learning_r": [7, 490, 491, 492, 493, 494, 495, 496, 497, 499, 501, 504, 505, 506, 507, 508, 509, 510, 517], "least": [6, 89, 90, 91, 101, 158, 204, 205, 207, 208, 209, 210, 211, 215, 216, 219, 220, 258], "leav": [2, 88, 146, 340, 341, 342], "lectur": 504, "lecun": 370, "left": [0, 6, 158, 201, 214, 258, 281, 374, 421, 435, 447, 448, 459, 461, 469], "left_shift": 0, "leibler": 462, "len": [6, 162, 166, 170, 173, 187, 508], "length": [6, 158, 167, 305, 361, 363, 366, 376, 382, 417, 508], "leq": [461, 475], "less": [0, 1, 6, 29, 203, 254, 293, 335, 421, 468, 498, 518], "less_equ": 0, "let": [1, 2, 5, 6, 205, 517, 519, 520, 523, 524], "level": [0, 179, 180, 439, 440], "lh": [376, 382, 417], "lhs_indic": [0, 179, 180], "lhs_mask": 96, "lib": [2, 518], "libbla": 9, "libcudnn9": 9, "liblapack": 9, "libmlx": 9, "libmlx_ext": 2, "libmpi": 518, "librari": [1, 2, 4, 9, 349, 356, 518, 519], "like": [2, 6, 8, 136, 150, 152, 190, 198, 251, 334, 371, 464, 501, 503, 517, 518, 519, 520, 522, 523, 524, 525, 527], "likelihood": [459, 467], "limit": [0, 2, 101, 289, 292, 293, 521], "linalg": 187, "line": [6, 518, 519, 522, 523, 524], "linear": [0, 2, 6, 7, 8, 217, 218, 336, 340, 356, 362, 372, 374, 375, 384, 399, 415, 417, 418, 420, 422, 424, 435, 444, 445, 446, 447, 448, 449, 453, 472, 474, 475, 477, 485, 488, 501, 508, 509, 517, 519], "linear1": 6, "linear2": 6, "linear3": 6, "linear_schedul": [490, 508], "linearli": 412, "link": [2, 4, 9], "linspac": 0, "lion": 490, "list": [1, 6, 11, 16, 18, 31, 75, 82, 87, 88, 89, 90, 91, 98, 102, 103, 110, 141, 146, 154, 156, 161, 162, 163, 165, 166, 167, 169, 170, 172, 173, 178, 184, 199, 214, 232, 234, 236, 237, 242, 250, 253, 256, 261, 262, 263, 265, 266, 267, 269, 272, 273, 285, 300, 302, 306, 307, 311, 318, 319, 322, 327, 328, 330, 333, 339, 342, 343, 356, 397, 399, 400, 401, 402, 407, 409, 410, 411, 488, 494, 495, 496, 497, 498, 508, 516, 517, 518, 519, 520, 522, 523], "list_or_scalar": 82, "listen": 518, "liter": [2, 253, 435, 439, 440, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469], "littl": 6, "liu": 6, "live": [8, 154, 156, 527], "ll": [1, 5, 7, 461, 517, 520], "llama": 6, "llamaattent": 6, "llamaencoderlay": 6, "llm": 8, "load": [2, 7, 8, 349, 399, 518], "load_weight": [356, 523], "loader": 7, "loader_path": 2, "loan": 214, "loc": [1, 265, 267], "local": [356, 370, 518], "localhost": [518, 522], "locat": [0, 2, 4, 87, 298, 299, 410, 411, 518, 521, 527], "log": [0, 225, 227, 232, 386, 387, 454, 455, 456, 459, 462, 464, 467, 480], "log10": 0, "log1p": 0, "log2": 0, "log_cosh_loss": 356, "log_sigmoid": [356, 386], "log_softmax": [356, 387], "logaddexp": 0, "logarithm": [0, 223, 224, 225, 226], "logcosh": 464, "logcumsumexp": 0, "logic": [0, 2, 229, 230, 231, 518], "logical_and": 0, "logical_not": 0, "logical_or": 0, "logist": [0, 5, 294, 448, 477], "logit": [6, 262, 456, 458, 517], "logsigmoid": 356, "logsoftmax": 356, "logsumexp": [0, 228], "long": 6, "longer": [6, 114, 520], "look": [2, 6, 518], "lookup": 373, "loop": [6, 7, 517, 518, 520, 523], "loshchilov": 495, "loss": [5, 7, 327, 356, 490, 517, 518, 520, 523], "loss_and_grad": 356, "loss_and_grad_fn": [7, 490, 517, 520], "loss_fn": [5, 7, 490, 517, 520], "loss_grad_fn": 518, "lot": 520, "low": [269, 273, 443, 485], "low_pad_s": 0, "lower": [204, 205, 208, 210, 218, 220, 258, 269, 272, 273, 323, 443], "lr": [5, 497], "lr_schedul": [506, 507, 508, 509, 510], "lstm": 356, "lto": 2, "lu": [6, 213], "luckili": 523, "lvalu": 327, "m": [0, 2, 4, 6, 9, 96, 153, 179, 187, 214, 323, 491, 517], "m1": [1, 6, 517, 520, 527], "m10": 347, "m7": 347, "m_": [494, 495, 496, 497], "m_t": [494, 495, 496, 497], "mac": 518, "machin": [6, 8, 9, 504, 518], "maco": [9, 293], "macosx": 9, "made": [6, 349], "mai": [2, 4, 88, 150, 152, 179, 180, 190, 214, 336, 370, 499, 518, 520, 521], "main": [4, 8, 128, 153, 154, 156, 321, 340, 341, 356, 518], "maintain": [370, 371, 497], "major": [0, 2, 106], "make": [1, 2, 3, 4, 6, 7, 9, 106, 151, 152, 233, 248, 291, 356, 498, 506, 507, 509, 510, 517, 523, 525, 527], "make_shar": 2, "malloc": 2, "man": 6, "manag": [8, 309, 514, 518, 519, 527], "mani": [1, 2, 87, 302, 363, 364, 365, 366, 367, 368, 373, 414, 517, 518, 519, 523], "manual": [356, 518], "map": [2, 7, 39, 222, 340, 373, 392, 519], "map_fn": [392, 396], "map_torch_to_mlx": 6, "margin": [465, 469], "margin_ranking_loss": 356, "mask": [0, 6, 96, 159, 406, 412, 521], "mask_lh": [0, 96], "mask_n": 1, "mask_nw": 1, "mask_out": [0, 96], "mask_rh": [0, 96], "mask_s": 1, "mask_sw": 1, "matadata": 222, "match": [9, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 181, 399, 435, 458, 521, 524], "materi": [6, 8], "math": [6, 469, 517], "mathbf": 205, "mathcal": [267, 385], "mathemat": 214, "mathrm": [144, 294, 381], "matmul": [0, 179, 527], "matric": [214, 216, 219], "matrix": [0, 5, 15, 46, 96, 126, 127, 153, 179, 180, 187, 188, 204, 205, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 219, 220, 233, 237, 258, 259, 266, 414, 415, 441, 485], "matter": [6, 356, 519], "matur": 518, "max": [0, 1, 2, 214, 235, 362, 388, 389, 390, 413, 444, 451, 452, 457, 459, 460, 465, 469, 471, 473, 474, 492, 496, 517, 520, 527], "max_buffer_s": 238, "max_freq": 426, "max_i": 258, "max_norm": 338, "max_recommended_working_set_s": [238, 293], "max_val": 451, "maximum": [0, 7, 27, 39, 101, 118, 183, 292, 338, 356, 384, 388, 389, 390, 418, 426, 447, 448, 453, 472, 488, 523], "maxpool1d": 356, "maxpool2d": 356, "maxpool3d": 356, "maxtotalthreadsperthreadgroup": 2, "mca": [518, 522], "md": 214, "me": 6, "mean": [0, 1, 5, 6, 7, 154, 157, 265, 266, 267, 327, 356, 361, 377, 397, 416, 442, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 517, 518, 520, 524], "meant": 122, "measur": 527, "mechan": 434, "medic": 371, "meet": 9, "member": [356, 402, 407], "memori": [0, 1, 2, 8, 87, 100, 154, 181, 182, 183, 278, 289, 292, 293, 434, 488, 492, 517, 523, 524], "memory_order_relax": 1, "memory_s": [238, 293], "memoryview": [523, 524], "merg": 517, "meshgrid": 0, "metadata": [5, 222, 285, 286], "metal": [2, 4, 8, 154, 156, 292], "metal_captur": 3, "metal_kernel": 1, "metal_path": 9, "metallib": [2, 9], "method": [2, 6, 10, 11, 31, 122, 130, 174, 336, 344, 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, 403, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 488, 491, 492, 493, 494, 495, 496, 497, 498, 499, 501, 504, 505, 512], "millisecond": [9, 517, 527], "min": [0, 2, 214, 243, 362, 413, 444, 451, 452, 471, 474], "min_freq": 426, "min_i": 258, "min_val": 451, "mind": 6, "mine": 6, "minibatch": 7, "minim": [518, 522], "minimum": [0, 28, 39, 101, 119, 426, 456, 457], "minsizerel": 9, "minu": 149, "minut": 6, "mish": 356, "mismatch": 519, "miss": [399, 519, 526], "mix": 521, "mkdir": [3, 9], "ml": 9, "mlp": [7, 356, 434, 490], "mlp_dim": [6, 434], "mlx": [1, 3, 5, 6, 7, 9, 349, 356, 485, 488, 490, 514, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "mlx_build_acceler": 4, "mlx_build_benchmark": 9, "mlx_build_cpu": 9, "mlx_build_cuda": 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": [129, 142, 517], "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": [150, 152, 190, 519], "mnist": 7, "mode": [0, 1, 2, 114, 126, 180, 253, 258, 259, 336, 395, 406, 408, 414, 415, 435, 439, 440], "model": [5, 7, 8, 287, 336, 337, 340, 341, 356, 392, 395, 397, 399, 403, 406, 408, 409, 410, 412, 434, 485, 488, 490, 500, 501, 503, 517, 518, 519, 523], "modest": 2, "modif": 524, "modifi": 524, "modul": [2, 4, 6, 7, 336, 337, 423, 434, 485, 503, 516, 517, 523], "moment": [6, 492, 496, 518], "momentum": [361, 497, 499, 501, 505, 517], "monei": 6, "monitor": 522, "monoton": 470, "moor": 215, "more": [1, 2, 3, 4, 7, 11, 82, 128, 150, 179, 204, 205, 207, 208, 209, 210, 211, 212, 215, 219, 220, 233, 258, 285, 286, 289, 292, 347, 356, 361, 370, 421, 426, 434, 435, 437, 438, 439, 440, 456, 514, 517, 518, 520, 521, 525, 527], "moreov": 522, "most": [2, 159, 262, 326, 356, 503, 517, 518, 520, 521, 523], "move": [0, 2, 244, 527], "moveaxi": 0, "mpi": [133, 349], "mpirun": [518, 522], "mse": 327, "mse_loss": 356, "mtl": 2, "mtl_capture_en": 3, "mtlcommandbuff": 2, "mu": 505, "much": [1, 2, 6, 358, 359, 360, 388, 389, 390, 517, 523], "multi": [8, 159, 363, 364, 365, 366, 367, 368, 519, 521, 524], "multidimension": 237, "multiheadattent": [6, 356], "multioptim": 490, "multipl": [0, 1, 9, 15, 96, 152, 155, 157, 179, 180, 233, 245, 259, 412, 426, 507, 508, 510, 517, 523, 526], "multipli": [0, 2, 39, 180, 259, 369, 426, 435], "muon": 490, "murtadha": 6, "must": [0, 1, 2, 3, 9, 96, 101, 150, 158, 159, 178, 180, 208, 210, 214, 258, 261, 262, 266, 269, 272, 273, 332, 435, 524], "mutat": 521, "mx": [1, 2, 3, 4, 5, 6, 7, 39, 88, 104, 105, 122, 133, 136, 150, 151, 152, 154, 156, 159, 175, 190, 198, 200, 207, 208, 209, 210, 212, 214, 216, 222, 268, 287, 298, 299, 326, 327, 338, 356, 358, 359, 360, 361, 372, 381, 384, 388, 389, 390, 392, 399, 403, 418, 435, 436, 437, 438, 439, 440, 441, 442, 443, 445, 453, 456, 457, 458, 462, 465, 472, 483, 485, 488, 490, 514, 517, 518, 519, 520, 521, 523, 524, 525, 526, 527, 528], "mxfp4": [126, 258], "my": [6, 9], "my_devic": 528, "my_path": 287, "my_script": [518, 522], "myexp": [1, 154, 156], "myexp_strid": 1, "mymlp": 488, "n": [0, 1, 2, 6, 31, 96, 107, 108, 109, 110, 111, 112, 113, 153, 159, 160, 162, 164, 166, 168, 171, 173, 187, 188, 266, 267, 307, 323, 328, 361, 363, 364, 365, 366, 367, 368, 370, 371, 376, 382, 417, 435, 464, 469, 518, 522], "n_kv": 159, "n_q": 159, "n_t": 376, "naiv": [2, 520], "naive_add": 520, "name": [1, 2, 122, 151, 154, 156, 180, 222, 259, 285, 286, 287, 288, 356, 377, 396, 399, 401, 518, 521, 526], "named_modul": 356, "namespac": 4, "nan": [0, 17, 86, 192, 193, 195, 246], "nan_to_num": 0, "nanobind": 2, "nanobind_add_modul": 2, "nativ": [9, 518], "natur": [0, 223, 225, 523], "nb": 2, "nb_domain": 2, "nb_modul": 2, "nb_static": 2, "nbyte": 2, "nc": 361, "nccl": 133, "ndarrai": [31, 521, 523, 525], "ndhwc": [365, 368, 371], "ndim": [0, 1, 2, 175, 214, 219, 435], "ne": 1, "nearest": [1, 435], "necessari": [106, 356], "necessarili": [209, 320], "need": [1, 2, 4, 6, 7, 8, 9, 86, 356, 410, 411, 426, 434, 514, 518, 520, 522, 523, 524, 525, 527], "neg": [0, 128, 175, 196, 246, 281, 321, 384, 388, 389, 390, 412, 459, 467, 469, 521], "negat": [0, 247], "negative_slop": [384, 453], "neginf": [0, 246], "neighbor": [435, 522], "neither": [184, 327], "nelem": 2, "nervou": 6, "nest": [82, 102, 342, 356, 488, 516, 520], "nesterov": [499, 505], "network": [6, 8, 335, 361, 370, 373, 437, 438, 485, 488, 499, 504, 518], "neural": [6, 8, 373, 437, 438, 470, 485, 488, 499, 504], "never": [6, 523], "new": [0, 1, 2, 7, 98, 128, 244, 248, 279, 306, 322, 329, 340, 341, 404, 412, 488, 490, 503, 508, 517, 519, 521, 523, 524], "new_tre": 341, "newton": 499, "next": [2, 4, 6, 7, 289, 519], "nh": [376, 382, 417], "nhwc": [361, 364, 367], "nice": [520, 523], "nlc": [361, 363, 366], "nld": [376, 382, 417], "nlh": [376, 382, 417], "nll": [459, 467], "nll_loss": 356, "nn": [2, 6, 7, 287, 340, 356, 485, 488, 490, 501, 503, 517, 519, 523], "nobodi": 6, "node": [88, 102, 146, 331, 341, 342, 518, 522], "nois": 5, "noisi": 5, "nomins": 2, "non": [0, 1, 2, 4, 9, 237, 407, 417, 470, 488], "nondeterminist": 521, "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, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 86, 87, 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, 118, 119, 120, 121, 122, 125, 126, 127, 128, 129, 131, 132, 135, 136, 137, 138, 139, 140, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 153, 155, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 184, 185, 186, 187, 188, 189, 191, 192, 193, 194, 195, 196, 197, 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, 240, 241, 242, 243, 244, 245, 246, 247, 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, 276, 277, 278, 279, 280, 281, 282, 283, 284, 286, 290, 291, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 331, 332, 333, 334, 335, 336, 339, 340, 341, 342, 358, 359, 360, 374, 388, 389, 390, 392, 396, 397, 404, 409, 412, 417, 426, 434, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 492, 512, 519, 521], "nonlinear": [417, 517], "nonzero": 521, "noop": [409, 518], "nor": [2, 184, 327], "norm": [6, 157, 338, 377, 469, 496, 497], "norm1": 6, "norm2": 6, "norm_first": 434, "normal": [1, 2, 5, 6, 154, 155, 156, 157, 159, 207, 208, 266, 272, 356, 358, 359, 360, 361, 377, 381, 383, 388, 389, 390, 416, 434, 437, 439, 524, 527], "not_equ": 0, "notabl": [6, 8], "notat": [126, 339, 401], "note": [0, 1, 2, 4, 6, 9, 17, 19, 87, 96, 102, 109, 112, 113, 122, 126, 169, 170, 180, 181, 192, 212, 214, 258, 262, 329, 336, 356, 416, 435, 490, 521, 524, 526], "noth": [6, 122, 356, 523], "notic": [6, 519, 520, 526], "now": [1, 2, 6, 9, 415, 517, 524], "np": [1, 6, 7, 518, 524, 525], "npy": [222, 284, 526], "npz": [6, 222, 287, 288, 399, 403, 526], "ns_step": 499, "nuc": 214, "nuclear": 214, "nuisanc": 518, "nullopt": 0, "num": [0, 6, 221, 271], "num_class": [7, 490], "num_decoder_lay": 434, "num_embed": [373, 414], "num_encoder_lay": 434, "num_epoch": [7, 490], "num_exampl": 5, "num_featur": [5, 361], "num_group": 377, "num_head": [6, 412, 434], "num_it": 5, "num_lay": [6, 7, 490], "num_param": 356, "num_paramet": 413, "num_sampl": 262, "num_split": 0, "number": [0, 2, 12, 19, 64, 74, 102, 108, 109, 110, 112, 113, 126, 151, 153, 159, 180, 184, 188, 199, 221, 246, 253, 258, 259, 262, 265, 267, 271, 273, 277, 281, 282, 318, 319, 323, 327, 330, 331, 335, 336, 356, 361, 363, 364, 365, 366, 367, 368, 370, 371, 377, 381, 412, 413, 434, 435, 437, 438, 439, 440, 499, 506, 508, 509, 514, 517, 520, 522, 528], "number_of_el": 0, "numer": [6, 155, 157, 214, 227, 232, 300, 361, 377, 381, 383, 416, 456, 457, 459, 469, 491, 492, 493, 494, 495, 496, 504, 517, 523], "numpi": [2, 6, 7, 8, 14, 17, 19, 92, 94, 95, 97, 98, 138, 139, 143, 185, 186, 192, 201, 202, 203, 207, 209, 227, 233, 235, 243, 245, 249, 255, 276, 280, 310, 523, 525, 526], "nvidia": 9, "nw": 1, "nwhc": 370, "o": [9, 159, 382], "o_t": 382, "obj": 285, "object": [3, 11, 31, 102, 150, 151, 152, 154, 156, 198, 331, 339, 340, 341, 342, 347, 370, 434, 516, 522], "observ": 6, "occupi": [126, 180, 258, 259], "occur": 524, "odd": 167, "odim": 7, "odot": [376, 382], "off": [6, 9, 523], "offer": 464, "offset": [0, 1, 2, 6, 47, 87, 128, 155, 158, 321], "often": 371, "ok": [399, 517, 519, 520], "okai": [517, 523], "old": 6, "older": [150, 152, 190], "omit": [496, 518], "onc": [1, 2, 9, 517, 519], "one": [0, 2, 4, 6, 9, 39, 82, 89, 97, 101, 108, 109, 110, 112, 113, 133, 148, 150, 153, 155, 157, 158, 179, 214, 225, 233, 259, 262, 305, 310, 326, 335, 347, 366, 367, 368, 409, 435, 458, 498, 518, 519, 522, 527], "ones": [0, 2, 6, 251, 287, 299, 323, 410, 411, 490, 518, 521], "ones_lik": 0, "onli": [1, 2, 6, 8, 86, 96, 108, 109, 110, 112, 113, 122, 159, 179, 208, 210, 214, 219, 258, 266, 293, 329, 347, 356, 396, 397, 399, 404, 406, 409, 410, 411, 488, 517, 518, 519, 520, 522, 526, 527], "onlin": 493, "op": [1, 2, 252, 329, 397, 523], "open": [3, 9, 19, 269, 273, 518], "openmpi": 518, "oper": [3, 6, 8, 10, 38, 89, 90, 91, 110, 159, 179, 180, 255, 257, 300, 308, 315, 344, 347, 356, 434, 497, 517, 518, 520, 521, 523, 524, 525, 527, 528], "operand": [140, 141, 179], "opportun": 517, "opt": [500, 518], "optim": [1, 3, 5, 7, 8, 410, 517, 518, 520, 523], "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, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 87, 89, 90, 91, 96, 102, 103, 107, 108, 109, 110, 111, 112, 113, 114, 118, 119, 120, 121, 122, 126, 127, 128, 131, 132, 133, 135, 136, 137, 150, 152, 153, 155, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 178, 179, 180, 184, 188, 196, 197, 200, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 228, 232, 234, 236, 237, 242, 246, 250, 253, 254, 256, 258, 259, 261, 262, 263, 265, 266, 267, 268, 269, 271, 272, 273, 277, 279, 281, 286, 300, 301, 302, 305, 306, 307, 311, 313, 314, 318, 320, 321, 322, 323, 324, 325, 326, 327, 328, 331, 333, 335, 336, 339, 340, 341, 342, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 376, 382, 385, 388, 389, 390, 392, 396, 397, 399, 404, 409, 412, 414, 415, 417, 421, 426, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 491, 492, 493, 494, 495, 496, 497, 499, 501, 504, 505, 506, 514, 517, 519, 526, 528], "ord": 214, "order": [0, 1, 29, 87, 110, 141, 208, 209, 210, 214, 254, 320, 356, 377, 410, 423, 501, 517, 520, 522], "ordinari": 191, "org": [377, 381, 383, 391, 416, 448, 470], "origin": [6, 128, 338, 361, 405, 437, 438, 439, 440, 491, 492, 493, 496, 497, 499, 519, 521, 524], "orthogon": 499, "orthonorm": 187, "other": [0, 2, 6, 8, 198, 214, 356, 398, 488, 497, 517, 518, 519, 521, 522, 523, 525], "other_input": 356, "otherwis": [19, 110, 133, 268, 336, 339, 340, 341, 342, 397, 399, 409, 432, 434, 435, 450, 456, 461, 468, 481, 482, 523, 524], "our": [1, 2, 6, 7, 423, 491, 492, 493, 496, 497, 499, 518], "out": [0, 1, 2, 9, 96, 154, 156, 159, 190, 370, 371, 406, 517, 518, 519, 520, 521], "out_ax": [331, 520], "out_channel": [363, 364, 365, 366, 367, 368], "out_dim": [356, 488], "out_dtyp": 2, "out_idx": 2, "out_mask": 96, "out_proj": [6, 488], "out_ptr": 2, "out_shap": [1, 2], "outer": [0, 517, 523], "outlier": 464, "output": [0, 1, 2, 6, 9, 16, 17, 18, 19, 29, 87, 96, 97, 98, 102, 104, 105, 106, 111, 112, 113, 118, 119, 120, 121, 122, 140, 151, 153, 154, 155, 156, 157, 158, 159, 168, 171, 172, 173, 178, 179, 184, 187, 188, 192, 214, 221, 228, 232, 234, 236, 237, 242, 246, 250, 251, 254, 256, 257, 261, 262, 263, 265, 266, 267, 269, 272, 273, 287, 288, 298, 299, 300, 305, 307, 311, 315, 321, 323, 327, 328, 329, 330, 331, 332, 333, 334, 361, 363, 364, 365, 366, 367, 368, 381, 385, 412, 415, 432, 434, 435, 437, 438, 439, 440, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 482, 485, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527], "output_dim": [7, 356, 385, 415], "output_directori": 2, "output_dtyp": [1, 154, 156], "output_fil": 6, "output_nam": [1, 154, 156], "output_pad": [0, 111, 112, 113, 366, 367, 368], "output_shap": [1, 154, 156], "output_strip_trailing_whitespac": 4, "output_vari": 4, "outsid": [154, 156, 175], "over": [0, 2, 6, 7, 16, 18, 27, 28, 29, 30, 107, 108, 109, 110, 111, 112, 113, 118, 119, 120, 121, 162, 163, 166, 167, 170, 173, 191, 214, 219, 221, 228, 232, 234, 236, 242, 254, 256, 283, 300, 301, 307, 311, 318, 320, 328, 361, 363, 364, 365, 366, 367, 368, 377, 383, 416, 458, 506, 509, 518, 520, 522], "overal": 2, "overhead": [1, 517, 523, 527], "overlap": 1, "overload": 19, "overrid": [2, 142], "overview": 3, "overwrit": 6, "own": [9, 518, 524], "owndata": 524, "p": [9, 212, 261, 356, 369, 370, 371, 469, 494, 496], "pack": [180, 258, 259], "packag": [2, 5, 7, 9, 349, 485, 518, 522], "package_data": 2, "pad": [0, 1, 107, 108, 109, 110, 111, 112, 113, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 358, 359, 360, 363, 364, 365, 366, 367, 368, 388, 389, 390], "pad_valu": 0, "pad_width": [0, 253], "padding_hi": 0, "padding_lo": 0, "page": [518, 525], "pain": 6, "pair": [0, 2, 253, 399, 421], "pairwis": 469, "pan": 6, "paper": [361, 426, 491, 492, 493, 496, 497], "parallel": [518, 527], "param": [327, 336, 356, 485, 519, 520], "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, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 125, 126, 127, 128, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 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, 175, 176, 177, 178, 179, 180, 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, 240, 242, 243, 244, 245, 246, 247, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 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, 335, 336, 337, 338, 339, 340, 341, 342, 343, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 381, 382, 383, 384, 385, 388, 389, 390, 392, 393, 396, 397, 399, 404, 405, 406, 409, 410, 411, 412, 413, 414, 415, 416, 417, 421, 423, 426, 430, 432, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 449, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 482, 484, 485, 488, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 503, 504, 505, 506, 507, 508, 509, 510, 512, 517, 518, 519, 520, 523], "parameter_scal": 492, "parametr": [413, 471], "pars": [6, 151], "parse_arg": 6, "parser": 6, "part": [1, 2, 51, 67, 150, 152, 189, 190, 274, 520, 521], "parti": 518, "partial": [410, 411, 517, 523], "particip": [131, 132, 135, 136, 137], "particular": [258, 377], "particularli": 517, "partit": [0, 29], "pass": [1, 2, 6, 7, 9, 69, 83, 179, 180, 252, 253, 327, 335, 337, 339, 340, 341, 356, 397, 409, 410, 411, 423, 517, 518, 519, 522, 523], "password": [518, 522], "path": [3, 4, 9, 141, 150, 151, 152, 190, 222, 240, 284, 285, 286, 287, 288, 336, 341, 399, 498, 518, 522], "pattern": [356, 523], "peak": [183, 278], "penalti": [499, 505], "penros": 215, "pep": 524, "per": [6, 7, 126, 159, 180, 258, 259, 335, 336, 361, 377, 381, 383, 416, 512, 517, 518, 522, 523], "perceptron": [8, 519], "perf_count": 517, "perfectli": 523, "perform": [0, 1, 2, 3, 6, 8, 15, 96, 110, 118, 119, 120, 121, 137, 140, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 179, 180, 187, 208, 210, 228, 233, 259, 282, 300, 314, 335, 356, 377, 434, 439, 440, 490, 499, 517, 518, 521, 523, 527], "perhap": [2, 6], "perm": 7, "permtuat": 268, "permuat": 212, "permut": [0, 7], "persist": 9, "pg": 214, "phi": [374, 446], "physic": 518, "pi": [144, 374, 426, 447, 520], "pick": 2, "pip": [2, 4, 9], "pipelin": 2, "pivot": [212, 213], "pixel": 370, "place": [6, 39, 281, 282, 336, 518, 523, 524], "placehold": 517, "plai": [2, 6], "plain": 423, "plan": [2, 517], "platform": 9, "plot": 518, "plu": [0, 225], "png": 518, "point": [0, 2, 5, 6, 9, 87, 174, 177, 258, 259, 347], "pool": [358, 359, 360, 388, 389, 390, 527], "popul": 2, "port": 522, "portion": 369, "posinf": [0, 246], "posit": [0, 6, 29, 128, 158, 175, 184, 197, 204, 205, 244, 246, 254, 266, 281, 321, 327, 340, 356, 363, 364, 365, 366, 367, 368, 412, 421, 426, 459, 469, 519], "possibl": [133, 179, 302, 373, 414, 517, 518, 521, 527], "possibli": [1, 6, 15, 96, 179, 233, 338], "postur": 6, "power": [0, 520, 524], "practic": [2, 517], "pre": [9, 159, 456], "preced": 377, "precis": [0, 6, 149, 159, 258, 356, 374, 416, 456, 500, 517], "preclud": 356, "pred": [460, 464], "predic": [336, 404, 498], "predict": [456, 459, 460, 461, 462, 463, 464, 466, 467, 468], "prefix": [331, 339, 341], "prelu": 356, "prepar": [2, 6, 518], "prepend": [3, 233], "preprint": [6, 491, 497], "preprocessor": 9, "present": 1, "preserv": [279, 520], "press": [6, 214], "pretti": [517, 523], "prevent": [308, 469, 524], "previou": [289, 292, 293], "primal": [1, 2, 122, 199, 330], "primit": 520, "print": [1, 2, 5, 6, 7, 9, 88, 122, 200, 338, 339, 340, 341, 343, 356, 514, 517, 518, 519, 520, 521, 522, 523, 524, 525], "prior": [257, 314, 315], "priorit": 520, "privat": [2, 4], "prng": [261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 514], "prob": 456, "probabl": [9, 269, 369, 370, 371, 415, 456, 458, 462, 527], "problem": [5, 7, 356], "process": [6, 110, 114, 130, 131, 132, 133, 135, 136, 137, 335, 340, 341, 371, 373, 434, 516, 518, 522], "processor": 9, "prod": [0, 1], "produc": [0, 2, 9, 102, 412, 485, 519], "product": [0, 2, 15, 87, 120, 191, 199, 200, 206, 233, 252, 256, 318, 330, 412, 525], "profil": 3, "program": [4, 183], "programmat": 411, "project": [3, 4, 6, 412, 519], "project_source_dir": 2, "promot": [2, 159], "promote_typ": 2, "promoted_dtyp": 2, "prompt": 6, "propag": [520, 521], "properti": [32, 39, 48, 51, 53, 64, 65, 67, 72, 74, 405, 408, 502, 518, 520], "proportion": 338, "protocol": 524, "provid": [0, 2, 6, 87, 150, 151, 184, 267, 268, 281, 318, 327, 335, 340, 342, 349, 356, 392, 397, 399, 409, 410, 411, 414, 415, 434, 435, 484, 488, 498, 518, 519, 526, 528], "pseudo": [215, 514], "pth": 6, "public": [2, 356], "pun": 0, "pure": [1, 122, 356, 490], "purpos": [1, 214, 518], "purs": 6, "push": 2, "push_back": 2, "put": [0, 1, 7, 257, 517, 518], "put_along_axi": [0, 212], "py": [2, 6, 9, 518, 522], "pypi": 9, "python": [1, 3, 4, 6, 52, 72, 82, 88, 146, 335, 339, 340, 341, 342, 343, 488, 500, 501, 503, 516, 518, 519, 520, 522, 524], "python_execut": 4, "python_requir": 2, "pytorch": [6, 8, 374, 377, 520], "pytorch_compat": 377, "q": [159, 216], "quantiz": [0, 126, 180, 222, 259, 414, 415], "quantized_matmul": 0, "quantizedembed": 356, "quantizedlinear": 356, "quarter": 6, "queri": [6, 159, 293, 412], "query_input_dim": 412, "query_proj": 6, "question": [6, 523], "queue": 3, "quick": [2, 8], "quit": [520, 524], "quotient": [0, 138, 139, 177], "r": [2, 6, 216, 327, 370, 376], "r_t": 376, "race": 527, "radian": [0, 125], "rag": 6, "rain": 6, "rais": [0, 6, 122, 214, 255, 302, 399, 519], "ram": [6, 292], "random": [1, 2, 3, 5, 6, 7, 8, 154, 156, 159, 358, 359, 360, 361, 381, 388, 389, 390, 399, 406, 517, 519, 520, 527, 528], "randomli": [5, 6, 268, 369, 370, 371], "rang": [0, 2, 3, 5, 6, 7, 9, 19, 175, 179, 221, 438, 440, 447, 448, 490, 506, 507, 508, 509, 510, 514, 517, 520, 523, 527], "rank": [0, 135, 136, 137, 465, 518, 522], "rate": [5, 490, 491, 492, 493, 494, 495, 496, 497, 499, 504, 505], "rather": [2, 520, 527], "ratio": [0, 25], "rceil": 96, "re": [7, 9, 267, 485], "reachabl": 518, "readabl": 3, "real": [0, 168, 169, 170, 171, 172, 173, 204, 205, 207, 208, 209, 210, 267], "realli": 383, "reason": [1, 6, 521], "reboot": 9, "receiv": [135, 136, 336, 508, 518, 524], "reciproc": [0, 283], "reclaim": 289, "recommend": [9, 292, 497, 499], "recompil": [102, 517], "reconstruct": 212, "record": [3, 183, 523], "recreat": [343, 490], "rectifi": [384, 418, 420, 439, 440, 453, 472, 474], "recurr": [376, 382, 417], "recurs": [151, 356, 396, 397, 402, 407, 409, 488], "recv": [136, 518], "reduc": [0, 1, 9, 16, 18, 27, 28, 132, 232, 234, 236, 242, 256, 307, 311, 328, 335, 342, 361, 434, 464], "reduct": [16, 18, 132, 232, 234, 242, 256, 342, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469], "redund": 520, "refer": [214, 381, 391, 405, 437, 438, 439, 440, 448, 470, 521], "reflect": [405, 517, 521, 524], "regard": 374, "regardless": [87, 159, 518], "regist": [2, 7], "regress": [8, 464], "regular": [39, 370, 470, 495, 517, 519, 521], "regularli": 2, "reimplement": 2, "rel": [17, 192, 492, 517, 518], "relative_step": 492, "releas": 4, "relev": 2, "reli": [1, 2], "relu": [356, 413, 434, 471, 485], "relu2": 356, "relu6": 356, "relu\u00b2": [419, 473], "remain": [0, 6, 293, 327, 341, 369, 370, 371, 518], "remaind": [0, 139], "remov": [0, 128, 233, 262, 305, 458], "rep": [0, 319], "repeat": [0, 319], "repeatedli": 5, "repetit": 277, "replac": [0, 6, 246, 410, 411, 434, 468], "replai": 3, "repli": 6, "repo": [5, 7, 9, 517], "report": [181, 292], "repres": [2, 6, 130, 133, 180, 465, 469, 524], "represent": [6, 213, 258, 329, 339, 343], "request": 154, "requir": [1, 2, 4, 6, 356, 518, 522, 523, 524], "requires_grad": 520, "rerun": [517, 523], "rescal": 338, "research": 8, "reset": 278, "reset_peak_memori": 183, "reshap": [0, 6, 214, 435, 517, 521], "resid": 293, "resolv": 2, "resourc": 2, "resource_limit": 238, "respect": [2, 5, 7, 122, 155, 157, 179, 180, 184, 258, 327, 340, 356, 361, 374, 377, 381, 383, 488, 518, 520, 522, 525], "respons": 2, "rest": [6, 158, 340, 341, 421, 522], "restart": 9, "restor": 281, "result": [0, 6, 15, 19, 39, 82, 87, 102, 151, 155, 157, 180, 200, 214, 233, 259, 266, 277, 292, 306, 340, 341, 342, 347, 426, 456, 517, 518, 520, 524], "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, 52, 72, 82, 86, 87, 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, 118, 119, 120, 121, 122, 125, 126, 127, 128, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 143, 144, 145, 147, 148, 149, 150, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 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, 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, 242, 243, 244, 245, 246, 247, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 277, 279, 280, 282, 283, 289, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 336, 337, 338, 339, 340, 341, 342, 343, 356, 376, 382, 392, 393, 394, 396, 397, 398, 399, 400, 401, 402, 406, 407, 409, 410, 411, 417, 436, 437, 438, 439, 440, 441, 442, 443, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 485, 488, 498, 500, 516, 517, 518, 519, 520, 521, 523, 524, 526, 527], "return_metadata": 222, "revers": [0, 2, 42, 43, 44, 45, 58, 87, 118, 119, 120, 121, 228, 322, 426], "rf": 9, "rfft": 168, "rfft2": 169, "rfftn": 170, "rho": 491, "rhs_indic": [0, 179, 180], "rhs_mask": 96, "right": [0, 1, 2, 9, 207, 258, 280, 281, 374, 435, 447, 448, 459, 461, 469], "right_shift": 0, "ring": 133, "rm": [6, 9, 157, 492], "rmsnorm": [6, 356], "rmsprop": 490, "rnn": [356, 376], "robust": 464, "roform": [6, 421], "roll": 0, "root": [0, 6, 157, 283, 303, 416], "rope": [6, 356], "rosetta": 9, "rotari": [6, 158, 421], "rotat": [158, 421], "round": [0, 258], "row": [0, 1, 2, 87, 106, 153, 154, 156, 188, 258, 323], "rpath": 2, "rsqrt": 0, "rtol": [0, 17, 192], "rule": [2, 490], "run": [1, 2, 3, 4, 6, 7, 8, 9, 10, 154, 156, 252, 344, 361, 392, 491, 492, 494, 495, 496, 517, 519, 522, 523, 527, 528], "runtim": [6, 133, 349, 517, 518], "runtime_error": 2, "safetensor": [9, 222, 286, 399, 403, 490, 523, 526], "sai": [2, 6, 485, 523], "said": 6, "sake": 520, "same": [0, 2, 6, 9, 17, 39, 86, 97, 98, 102, 108, 109, 110, 112, 113, 114, 131, 155, 157, 163, 167, 168, 171, 172, 173, 180, 184, 192, 199, 253, 262, 281, 282, 299, 329, 330, 332, 335, 341, 356, 359, 360, 361, 369, 377, 381, 389, 390, 414, 436, 437, 438, 439, 440, 441, 442, 443, 458, 469, 488, 500, 514, 517, 518, 519, 521, 522, 527], "sampl": [2, 5, 6, 221, 261, 262, 263, 265, 266, 269, 272, 273, 437, 438, 439, 440, 442, 443, 459, 465, 469, 514, 517, 519], "sat": 6, "save": [3, 6, 8, 222, 240, 258, 285, 286, 287, 288, 403, 519, 523], "save_gguf": 526, "save_safetensor": [403, 490, 526], "save_weight": 356, "savez": [6, 403, 526], "savez_compress": 526, "saw": [6, 520], "scalar": [0, 2, 14, 15, 17, 31, 52, 82, 86, 92, 93, 94, 95, 96, 98, 101, 138, 139, 143, 158, 177, 178, 184, 185, 186, 187, 192, 201, 202, 203, 221, 227, 229, 230, 231, 233, 235, 243, 245, 246, 249, 253, 255, 261, 267, 269, 272, 273, 276, 280, 285, 310, 327, 329, 332, 337, 469, 519, 520, 523, 525], "scale": [0, 2, 6, 15, 126, 155, 157, 158, 159, 180, 187, 258, 259, 265, 267, 338, 370, 371, 383, 412, 421, 422, 426, 435, 475, 492], "scale_arr": 2, "scale_factor": 435, "scale_paramet": 492, "scatter": 0, "scatter_add": 0, "scatter_add_axi": 0, "scatter_max": 0, "scatter_min": 0, "scatter_prod": 0, "schedul": [2, 490, 506, 507, 508, 509, 510, 512, 527], "schema": [3, 522], "schulz": 499, "scipi": [187, 212], "scope": 356, "score": [6, 159, 465], "script": [518, 522], "sdk": 9, "se": 1, "second": [6, 9, 128, 198, 200, 201, 229, 231, 233, 258, 280, 312, 321, 327, 359, 360, 389, 390, 457, 465, 492, 496, 517, 519, 520, 527], "second_layer_a": 523, "second_layer_b": 523, "secret": 6, "section": [1, 6, 9, 302, 469, 517, 518, 520], "see": [1, 2, 4, 6, 7, 9, 11, 12, 33, 34, 35, 36, 37, 40, 41, 42, 43, 44, 45, 47, 49, 50, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 214, 285, 286, 289, 336, 347, 356, 361, 362, 370, 372, 374, 378, 379, 380, 386, 387, 395, 413, 414, 415, 418, 419, 420, 421, 422, 424, 426, 427, 428, 429, 430, 431, 433, 435, 437, 438, 439, 440, 446, 447, 448, 475, 517, 518, 519, 520, 521, 522, 525, 527], "seed": 264, "seen": [518, 524], "segment": 0, "segmented_mm": 0, "select": [0, 3, 9, 208, 210, 320, 332, 392, 396, 404, 522], "self": [6, 7, 10, 31, 32, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 52, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 122, 174, 356, 470, 488], "selu": 356, "semant": [14, 92, 94, 95, 97, 98, 138, 139, 143, 185, 186, 201, 202, 203, 227, 233, 235, 243, 245, 249, 255, 276, 280, 310, 527], "semi": [204, 205, 266], "send": 518, "sender": 518, "sennrich": 6, "sensit": 464, "sentencepiec": 6, "separ": [0, 6, 69, 83, 377, 465], "sequenc": [6, 16, 18, 34, 35, 59, 60, 61, 62, 66, 78, 79, 80, 84, 87, 98, 110, 135, 148, 154, 156, 158, 161, 162, 163, 165, 166, 167, 169, 170, 172, 173, 178, 184, 232, 234, 236, 242, 250, 256, 261, 262, 263, 265, 266, 267, 269, 272, 273, 279, 298, 299, 300, 302, 305, 307, 311, 318, 319, 322, 326, 327, 328, 333, 361, 363, 366, 376, 382, 417, 434, 514, 527], "sequenti": [356, 485], "seri": 9, "serial": 490, "set": [2, 4, 6, 7, 9, 102, 122, 129, 131, 132, 133, 135, 136, 137, 142, 155, 157, 158, 238, 289, 290, 291, 292, 293, 309, 335, 374, 383, 385, 395, 397, 404, 405, 406, 409, 410, 415, 421, 432, 457, 469, 482, 488, 490, 492, 494, 495, 501, 514, 519, 520, 523], "set_byt": 2, "set_compute_pipeline_st": 2, "set_data": 2, "set_dtyp": 356, "set_input_arrai": 2, "set_memory_limit": 289, "set_output_arrai": 2, "set_vector_byt": 2, "setup": [2, 4, 5, 7, 9, 517, 518, 519], "sever": [6, 9, 107, 108, 109, 110, 111, 112, 113, 287, 288, 335, 517, 518, 522, 526], "sgd": [5, 7, 490, 497, 501, 506, 507, 510, 517], "shade": [1, 2], "shall": 6, "shape": [0, 2, 3, 6, 7, 69, 86, 87, 96, 97, 98, 102, 107, 108, 109, 110, 111, 112, 113, 128, 131, 135, 136, 150, 152, 154, 156, 158, 159, 160, 163, 164, 167, 168, 171, 172, 173, 178, 179, 187, 199, 211, 220, 233, 250, 251, 261, 262, 263, 265, 266, 267, 269, 272, 273, 279, 281, 299, 326, 329, 330, 332, 333, 334, 356, 358, 359, 360, 361, 363, 364, 365, 366, 367, 368, 370, 371, 376, 381, 382, 385, 388, 389, 390, 399, 417, 436, 437, 438, 439, 440, 441, 442, 443, 458, 469, 490, 517, 519, 520, 521, 525, 527], "shapeless": [0, 102, 150, 152], "share": [8, 126, 154, 180, 258, 259, 329, 518], "shared_memori": 154, "shazeer": 6, "shift": [0, 163, 167, 201, 280, 281, 361], "shop": 6, "should": [1, 2, 4, 5, 6, 7, 9, 87, 100, 128, 131, 154, 155, 156, 157, 159, 199, 240, 257, 258, 293, 315, 321, 327, 330, 335, 336, 339, 356, 363, 364, 365, 366, 367, 368, 370, 371, 406, 412, 423, 458, 460, 465, 488, 498, 499, 516, 517, 518, 519, 520, 523, 524, 528], "show": [9, 347, 517], "shown": 2, "shuffl": 7, "side": [0, 253, 358, 359, 360, 366, 367, 368, 388, 389, 390, 517], "sigma": [374, 375, 376, 382, 425, 437, 438, 439, 440, 448, 449, 454, 476, 477], "sigmoid": [0, 6, 356, 386, 424, 448, 454, 456, 477], "sign": [0, 17, 192, 347, 497], "signal": [114, 435], "signatur": [1, 154, 156], "signedinteg": [12, 198], "signific": 258, "significantli": 518, "silent": [171, 172, 173], "silicon": [2, 6, 8, 9, 527], "silu": 356, "sim": 267, "simd": 1, "simd_sum": 1, "simdgroup": 1, "simdgroup_s": 1, "similar": [6, 180, 198, 340, 410, 411, 412, 457, 518, 524, 526], "similarli": [2, 9, 233, 258, 520, 523], "simpl": [2, 6, 7, 356, 373, 484, 490, 517, 518, 519, 520, 522, 523], "simple_axpbi": 2, "simple_tim": 2, "simplest": [2, 356, 518], "simpli": [2, 6, 9, 372, 384, 418, 445, 453, 472, 483, 488, 517, 518, 520, 522], "simplic": 0, "simplifi": 518, "simultan": 1, "sin": [0, 122, 426, 519, 520, 525], "sinc": [1, 2, 6, 7, 180, 183, 488, 497, 508, 517, 519, 524, 527], "sine": [0, 22, 23, 296, 297, 519, 520], "sing": 214, "singer": 493, "singl": [7, 88, 146, 199, 222, 237, 253, 330, 359, 360, 389, 390, 517, 519, 521, 526], "singleton": [0, 16, 18, 27, 28, 133, 232, 233, 234, 236, 242, 256, 307, 311, 328, 518], "singular": [214, 215, 219], "sinh": 0, "sink": 159, "sinusoid": 426, "sinusoidalpositionalencod": 356, "size": [0, 1, 2, 6, 7, 53, 72, 96, 108, 109, 112, 113, 126, 148, 154, 155, 156, 157, 158, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 178, 180, 182, 187, 188, 198, 206, 214, 258, 259, 262, 279, 292, 293, 298, 302, 305, 326, 329, 335, 336, 356, 358, 359, 360, 363, 364, 365, 366, 367, 368, 373, 381, 388, 389, 390, 414, 415, 435, 492, 518, 523, 524], "size_in_megabyt": 293, "size_t": [0, 2], "skip": [3, 87], "slice": [0, 299, 521], "slice_s": [0, 298], "slice_upd": 0, "slide": [358, 359, 360, 388, 389, 390], "slight": [6, 523], "slightli": [421, 527], "slope": 384, "slow": 517, "slowli": 6, "sm": 9, "small": [6, 149, 155, 157, 335, 361, 377, 383, 416, 459, 464, 469, 517, 527], "smaller": [0, 9, 254, 335, 497, 517], "smallest": 214, "smile": 6, "smooth": [458, 468, 504], "smooth_l1_loss": 356, "sned": 137, "snippet": 518, "so": [1, 2, 6, 9, 184, 187, 327, 369, 435, 490, 517, 518, 521, 523, 527], "socket": 518, "softmax": [0, 6, 159, 356, 387, 455, 458], "softmin": 356, "softplu": [356, 391, 470], "softshrink": 356, "softsign": 356, "solut": [217, 218], "solv": 356, "some": [0, 2, 5, 6, 7, 151, 397, 409, 490, 501, 517, 518, 519, 520, 522, 523], "someon": 6, "someth": [5, 6, 521], "sometim": 517, "sonoma": 9, "soon": 6, "sort": [0, 29, 30, 179, 180, 254, 320], "sorted_indic": [0, 179, 180], "sourc": [0, 1, 2, 3, 4, 63, 135, 136, 154, 156, 244, 322, 518], "space": [0, 2, 221, 456, 467], "spars": [0, 237], "spatial": [108, 109, 110, 112, 113, 358, 359, 360, 377, 388, 389, 390, 435], "speak": [6, 214], "specif": [1, 2, 9, 258, 518, 520], "specifi": [0, 2, 19, 38, 108, 109, 110, 112, 113, 128, 169, 170, 178, 184, 206, 214, 221, 244, 250, 257, 262, 277, 312, 314, 315, 318, 321, 322, 327, 331, 333, 361, 432, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 482, 517, 518, 519, 520, 527], "spectrum": 163, "speed": 1, "spent": 6, "split": [0, 375, 377, 449], "splittabl": 514, "sqrt": [0, 6, 144, 159, 187, 361, 374, 377, 381, 383, 385, 416, 426, 437, 438, 439, 440, 447, 491, 493, 494, 495, 504, 517], "squar": [0, 5, 6, 157, 188, 207, 209, 211, 220, 283, 303, 327, 340, 356, 416, 466, 468, 491, 492, 494, 495, 496, 520, 524], "squeez": [0, 435, 517], "src": [0, 135, 136], "ssh": [518, 522], "stabil": [155, 157, 361, 377, 381, 383, 416, 456, 457, 459, 491, 492, 493, 494, 495, 496, 504], "stabl": [227, 232, 300, 464], "stable_abi": 2, "stack": [0, 517], "standard": [0, 1, 4, 7, 52, 82, 233, 263, 267, 307, 434, 437, 439, 442, 518, 525], "starmap": [6, 340], "start": [0, 1, 2, 5, 6, 8, 9, 19, 158, 221, 240, 298, 299, 302, 342, 517, 519, 521, 522, 527], "start_axi": [0, 50, 175], "start_captur": 3, "start_indic": [298, 299], "state": [6, 7, 356, 376, 382, 417, 490, 501, 514, 517], "static": [9, 517], "static_cast": 2, "std": [0, 2, 4, 442, 519], "stderr": 522, "stdout": 522, "step": [0, 3, 4, 6, 7, 19, 335, 356, 376, 382, 417, 492, 499, 501, 506, 508, 509, 510, 517, 518], "step_decai": 490, "step_siz": 510, "still": [6, 9, 214, 517, 523], "stochast": [493, 494, 496, 505, 523], "stood": 6, "stop": [0, 2, 6, 19, 221, 241, 308, 520, 521], "stop_captur": 3, "stop_gradi": [0, 520], "storag": 87, "store": [6, 339], "str": [114, 126, 133, 140, 141, 150, 151, 152, 154, 156, 159, 180, 184, 190, 208, 210, 214, 222, 237, 238, 240, 258, 259, 284, 285, 286, 287, 288, 327, 336, 339, 343, 392, 393, 396, 397, 399, 401, 403, 409, 414, 415, 435, 439, 440, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 498], "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, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 124, 125, 126, 127, 128, 131, 132, 135, 136, 137, 138, 139, 140, 143, 144, 145, 147, 148, 149, 153, 155, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 176, 177, 178, 179, 180, 185, 186, 187, 188, 189, 191, 192, 193, 194, 195, 196, 197, 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, 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, 271, 272, 273, 274, 275, 276, 277, 279, 280, 281, 282, 283, 291, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 328, 329, 332, 333, 334, 335, 518, 527], "streamcontext": 309, "streamordevic": [0, 2], "street": 6, "strength": [497, 499, 505], "strict": [133, 185, 202, 397, 399, 409, 410, 411], "strictli": [214, 293], "stride": [0, 2, 87, 107, 108, 109, 110, 111, 112, 113, 358, 359, 360, 363, 364, 365, 366, 367, 368, 388, 389, 390, 421, 521], "string": [0, 2, 141, 150, 154, 156, 159, 190, 238, 253, 519, 524, 526], "stronger": 522, "structur": [2, 335, 500, 520], "stub": 9, "style": [2, 14, 17, 92, 94, 95, 138, 139, 143, 185, 186, 192, 201, 202, 203, 227, 233, 235, 243, 245, 249, 255, 276, 280, 310], "su": 6, "sub": [0, 7, 128, 271, 298, 299, 321, 336, 499], "subarrai": [128, 302], "subclass": 488, "subdivid": 1, "subdtyp": 198, "subgradi": 493, "sublinear": 492, "submodul": [6, 7, 356, 393, 397, 398, 409, 411], "subnetwork": 518, "suboptim": 519, "subscript": [140, 141], "subsect": 6, "subsequ": [133, 490, 518, 522], "subset": [356, 396, 410, 411], "substanti": 9, "subtl": 517, "subtract": [0, 39], "subtyp": [198, 347], "succe": 133, "successfulli": 518, "sudo": [9, 293, 518], "suggest": 518, "sum": [0, 2, 5, 14, 121, 132, 152, 191, 214, 232, 300, 318, 321, 356, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 518, 521, 524], "sum_": [214, 464], "sum_i": 455, "sum_j": [478, 479], "summat": [140, 141], "super": [6, 7, 356, 488], "superset": [340, 500], "support": [1, 2, 6, 8, 9, 17, 96, 109, 112, 113, 126, 159, 175, 187, 192, 204, 205, 207, 208, 209, 210, 211, 215, 216, 219, 220, 222, 233, 258, 266, 518, 520, 521, 524, 526], "suppos": [520, 527], "sure": [2, 3, 6, 9, 356, 517], "surpass": [439, 440], "surpris": 6, "sw": 1, "swap": [0, 114, 292, 312, 411], "swapax": [0, 122], "swiglu": 6, "swish": [424, 477], "switch": 9, "symbol": 497, "symmetr": [108, 109, 112, 113, 204, 205, 208, 210], "symmetri": [208, 210], "synchron": [2, 517], "syntax": [39, 521], "synthet": 5, "sysctl": 293, "system": [4, 6, 9, 181, 182, 217, 218, 238, 293], "t": [0, 1, 2, 4, 6, 9, 106, 144, 154, 156, 158, 159, 180, 204, 205, 259, 327, 356, 376, 382, 417, 491, 492, 493, 494, 495, 496, 497, 504, 505, 517, 519, 520, 527], "t_kv": 159, "t_q": 159, "tabl": [1, 214, 347, 373], "take": [0, 2, 6, 7, 92, 93, 94, 95, 102, 150, 179, 184, 199, 235, 243, 251, 259, 315, 327, 330, 331, 334, 341, 342, 358, 359, 360, 388, 389, 390, 412, 456, 498, 514, 518, 519, 520, 521, 522, 526, 527, 528], "take_along_axi": [0, 212, 521], "taken": [128, 314, 321], "talk": 518, "tan": 0, "tangent": [0, 2, 24, 25, 26, 122, 199, 316, 317, 433, 483], "tangent_i": 2, "tangent_x": 2, "tanh": [0, 356, 374, 376, 382, 391, 417, 447, 470], "target": [2, 327, 456, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 517], "target_include_directori": 2, "target_link_librari": [2, 4], "target_link_opt": 2, "target_sourc": 2, "task": 464, "tau": 505, "tcp": 518, "tediou": 518, "tell": [4, 6, 517, 524], "temp": 6, "templat": [0, 1, 2, 154, 156], "ten": 523, "tend": 497, "tensor": [222, 318, 469, 524], "tensordot": 0, "term": [2, 459, 491, 492, 493, 494, 495, 496, 504], "termin": [9, 522], "test": [7, 9, 518, 522], "test_imag": 7, "test_label": 7, "text": [6, 267, 374, 376, 382, 391, 417, 425, 432, 437, 438, 439, 440, 447, 450, 451, 452, 459, 460, 461, 464, 465, 468, 470, 471, 475, 476, 481, 482, 492, 497], "textrm": [258, 374, 375, 446, 449], "tf": 524, "tgp_size": 2, "th": [118, 119, 120, 121, 127, 153, 207, 208, 228, 508], "than": [1, 2, 6, 82, 114, 128, 139, 158, 179, 185, 186, 202, 203, 204, 205, 207, 208, 209, 210, 211, 212, 215, 219, 220, 233, 289, 293, 338, 340, 421, 432, 435, 465, 468, 482, 492, 497, 498, 517, 519, 520, 527], "thank": 523, "thei": [1, 2, 5, 6, 9, 17, 114, 180, 192, 258, 423, 460, 488, 497, 516, 517, 518, 519, 523, 525, 526, 527], "them": [0, 2, 6, 131, 356, 397, 409, 518, 519, 522, 527], "themselv": [2, 517], "thi": [0, 1, 2, 4, 6, 7, 9, 16, 17, 18, 19, 27, 28, 29, 30, 87, 88, 100, 122, 142, 150, 152, 154, 156, 179, 180, 181, 187, 190, 192, 199, 204, 205, 207, 208, 209, 210, 211, 214, 215, 216, 219, 220, 227, 232, 233, 234, 236, 242, 254, 256, 262, 291, 293, 300, 301, 302, 307, 311, 314, 320, 328, 335, 338, 341, 342, 356, 369, 370, 371, 375, 376, 382, 393, 394, 396, 397, 400, 401, 402, 407, 409, 410, 411, 412, 415, 417, 432, 437, 438, 439, 440, 447, 448, 449, 456, 464, 482, 488, 498, 501, 516, 517, 518, 519, 520, 522, 523, 524, 526], "thin": 522, "thing": [2, 6], "third": [206, 360, 390, 518, 519], "this_grid": 154, "thompson": 370, "those": [2, 6, 356, 499], "though": [2, 6, 517, 519, 523, 524], "thousand": 523, "thread": [1, 2], "thread_index_in_simdgroup": 1, "thread_position_in_grid": [1, 2, 156], "thread_rank": 154, "threadgroup": [1, 2, 154, 156], "threads_per_simdgroup": 1, "three": [6, 91, 258, 360, 390, 435], "threefri": 514, "threshold": [432, 461, 468, 482], "through": [1, 2, 308, 434, 497, 517, 518, 519, 520, 524], "throw": [2, 102, 133], "thu": [6, 356], "thumb": 490, "tic": 517, "tieleman": 504, "tile": [0, 159], "time": [1, 2, 6, 9, 292, 319, 356, 376, 382, 417, 517, 518, 520, 523, 527], "timeit": [517, 520], "titl": 2, "tmp": [1, 154, 156], "to_quant": 336, "to_stream": 2, "toc": 517, "togeth": [0, 1, 2, 7, 258, 340, 341, 518], "tok_embed": 6, "token": [6, 373, 414], "told": 6, "toler": [0, 17, 192], "too": [198, 517, 523], "took": 6, "tool": 9, "toolkit": 9, "top": [2, 320, 385, 435], "topk": 0, "torch": [6, 524], "torch_weight": 6, "total": [293, 520], "total_norm": 338, "tpi": 517, "tpng": 518, "trace": [0, 3, 152, 517], "trace_fil": 3, "tracer": 410, "track": [2, 356, 361], "track_running_stat": 361, "trade": 523, "tradit": [6, 158, 370, 371, 421], "train": [6, 7, 356, 361, 369, 370, 371, 395, 397, 409, 437, 438, 519], "train_imag": [7, 490], "train_label": [7, 490], "trainabl": [7, 337, 356, 488], "trainable_paramet": [356, 396, 501], "transfer": 522, "transform": [1, 6, 8, 122, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 187, 337, 356, 361, 377, 383, 385, 396, 397, 409, 415, 421, 521], "transformerencod": 287, "transit": 508, "translat": [155, 383], "transpos": [0, 6, 32, 111, 112, 113, 180, 259, 366, 367, 368], "treat": [0, 2, 122, 169, 170, 172, 173, 314, 435, 517], "tree": [8, 88, 102, 146, 184, 327, 331, 335, 339, 340, 341, 342, 343, 500, 501, 503, 512, 520], "tree_flatten": [287, 340, 343, 356, 490, 519], "tree_map": [341, 356, 518], "tree_unflatten": [6, 490, 519], "trembl": 6, "tri": [0, 133], "triangl": [208, 210, 323], "triangular": [204, 205, 218, 220], "trigger": 517, "tril": 0, "trilinear": 435, "triplet": 469, "triplet_loss": 356, "triu": 0, "true": [0, 1, 2, 4, 5, 6, 17, 42, 43, 44, 45, 58, 86, 102, 118, 119, 120, 121, 154, 156, 158, 180, 192, 198, 204, 205, 214, 219, 222, 228, 237, 259, 300, 332, 336, 339, 340, 341, 342, 347, 356, 361, 363, 364, 365, 366, 367, 368, 376, 377, 381, 382, 383, 385, 396, 397, 399, 406, 409, 410, 411, 415, 417, 421, 426, 434, 435, 456, 464, 492, 494, 495, 498, 499, 517, 519], "truncat": [160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 272], "truth": [5, 458, 468], "try": [2, 9, 518], "tupl": [0, 31, 69, 72, 75, 83, 88, 97, 103, 108, 109, 110, 112, 113, 135, 139, 141, 146, 148, 150, 161, 162, 165, 166, 169, 170, 172, 173, 190, 199, 207, 208, 212, 213, 214, 216, 219, 253, 258, 279, 281, 298, 299, 305, 326, 327, 330, 339, 340, 341, 342, 343, 358, 359, 360, 364, 365, 367, 368, 388, 389, 390, 399, 401, 423, 435, 492, 494, 495, 496, 497, 516, 519, 520], "tutori": 2, "twice": 527, "two": [0, 2, 14, 15, 17, 25, 86, 90, 92, 94, 95, 96, 128, 138, 143, 161, 165, 172, 179, 180, 185, 186, 192, 200, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 215, 216, 219, 220, 227, 233, 235, 243, 245, 249, 252, 258, 312, 342, 359, 375, 382, 389, 449, 457, 517, 518, 519, 520, 521, 527], "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, 72, 82, 86, 87, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 118, 119, 120, 121, 125, 126, 127, 128, 131, 132, 133, 135, 136, 137, 138, 139, 140, 141, 143, 144, 145, 147, 148, 149, 153, 155, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 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, 242, 243, 244, 245, 246, 247, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 271, 272, 273, 274, 275, 276, 277, 279, 280, 282, 283, 289, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 310, 311, 312, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 338, 339, 342, 356, 404, 434, 436, 437, 438, 439, 440, 441, 442, 443, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 517, 519, 521, 524], "type_to_nam": 2, "typenam": [0, 1, 2], "typic": [0, 159, 335, 373, 490, 517, 523], "u": [1, 2, 4, 204, 208, 210, 212, 219, 385, 411, 512, 518, 522, 523], "u_": 491, "u_t": 491, "ubuntu": 9, "ubuntu2204": 9, "uint": [1, 2, 156], "uint16": [12, 347], "uint3": 1, "uint32": [12, 27, 28, 29, 30, 262, 347], "uint64": [12, 347], "uint8": [12, 347], "ultra": 6, "unabl": 9, "unam": 9, "unari": 517, "unchang": [158, 308, 421], "uncheck": 9, "uncompress": 287, "undefin": [0, 29, 122, 204, 205, 254, 266, 521], "under": 214, "underli": [2, 329], "understand": [6, 437, 438], "unevalu": 151, "unexpect": [2, 19], "unexpectedli": 522, "unflatten": 0, "unfreez": [356, 397], "unfrozen": 409, "unifi": 8, "uniform": [3, 356, 385, 399, 438, 440, 485, 514, 517, 520, 527], "uniformli": 273, "unintend": 0, "union": [19, 33, 34, 35, 36, 37, 38, 40, 41, 42, 43, 44, 45, 46, 47, 49, 50, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 66, 68, 69, 70, 71, 73, 75, 76, 77, 78, 79, 80, 81, 83, 84, 85, 89, 90, 91, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 196, 197, 198, 200, 209, 210, 219, 238, 285, 309, 339], "uniqu": [2, 217, 218, 514, 518], "unique_ptr": 2, "unit": [362, 372, 374, 375, 376, 384, 418, 420, 422, 424, 437, 438, 439, 440, 444, 445, 446, 447, 448, 449, 453, 472, 474, 475, 477], "unittest": 9, "univers": 214, "unless": [6, 17, 192, 214, 488], "unlik": [6, 17, 192, 212, 258, 370, 371, 405, 521], "unnecessari": [2, 6], "unnorm": [262, 456, 458], "unscal": 492, "unsign": [180, 258, 259, 347], "unsignedinteg": 12, "unspecifi": [16, 18, 19, 27, 28, 29, 30, 103, 118, 119, 120, 121, 178, 228, 232, 234, 236, 242, 250, 254, 256, 277, 300, 301, 307, 311, 314, 320, 321, 328, 333, 335, 528], "unsqueez": 6, "unsupport": 222, "until": [2, 335, 523, 525], "unus": 2, "up": [1, 2, 6, 122, 517], "upcast": 2, "updat": [0, 1, 2, 5, 6, 7, 9, 39, 102, 299, 336, 340, 342, 361, 392, 393, 399, 404, 405, 406, 411, 490, 492, 495, 497, 500, 501, 505, 506, 507, 508, 509, 510, 517, 518, 519, 523], "update_modul": 356, "uplo": [208, 210], "upon": [6, 340, 341], "upper": [204, 205, 208, 210, 218, 220, 258, 269, 272, 273, 443], "upsampl": 356, "us": [0, 3, 5, 6, 7, 8, 9, 19, 39, 87, 122, 126, 129, 131, 132, 135, 136, 137, 139, 154, 156, 158, 175, 180, 181, 182, 183, 201, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 233, 258, 259, 267, 277, 279, 280, 281, 289, 292, 293, 313, 326, 335, 336, 339, 342, 347, 349, 356, 359, 360, 370, 373, 374, 376, 382, 385, 389, 390, 392, 396, 403, 410, 412, 414, 415, 417, 421, 426, 434, 435, 439, 440, 447, 448, 457, 485, 488, 490, 491, 492, 494, 495, 496, 497, 498, 500, 501, 514, 516, 517, 518, 519, 520, 521, 522, 525, 527], "usag": [434, 517, 518], "user": [2, 6, 356], "uss": 335, "usual": [373, 414, 516, 523], "util": [1, 2, 6, 8, 9, 287, 356, 490, 522], "v": [6, 114, 159, 207, 208, 356, 397, 524], "v_": [491, 493, 494, 495, 496, 504, 505], "v_t": [491, 493, 494, 495, 496, 504, 505], "val": [0, 31, 178], "valid": [7, 114, 175, 331, 339, 397, 409, 516, 518], "valid_parameter_filt": 392, "valu": [0, 1, 5, 6, 12, 13, 17, 19, 27, 28, 52, 82, 86, 101, 133, 150, 153, 154, 159, 160, 161, 162, 164, 165, 166, 168, 169, 170, 171, 172, 173, 175, 178, 190, 192, 206, 214, 215, 219, 221, 238, 246, 253, 257, 258, 261, 262, 263, 265, 266, 267, 269, 272, 273, 281, 285, 293, 314, 315, 327, 331, 337, 339, 340, 341, 342, 347, 359, 360, 362, 369, 370, 371, 372, 378, 381, 385, 389, 390, 396, 412, 413, 430, 432, 434, 436, 456, 457, 458, 459, 460, 461, 463, 464, 465, 466, 467, 468, 482, 488, 492, 495, 506, 507, 509, 510, 520], "value_and_grad": [7, 122, 356, 410, 488, 490, 503, 517, 520, 524, 525], "value_and_grad_fn": 523, "value_cach": 6, "value_dim": 412, "value_input_dim": 412, "value_output_dim": 412, "value_proj": 6, "valueerror": [122, 214, 399, 520], "values_hat": 6, "van": 214, "var": [0, 361, 377, 381, 383, 459], "vari": 335, "variabl": [9, 102, 122, 129, 142, 150, 151, 152, 184, 199, 327, 330, 331, 517, 518, 519], "varianc": [0, 307, 328, 361, 377, 459], "variant": [6, 468, 496], "variou": 214, "vector": [0, 2, 5, 8, 158, 191, 199, 214, 314, 330, 331, 373, 458, 519, 525], "verbos": [1, 154, 156, 518], "veri": [6, 412, 522, 523, 527], "verifi": [5, 9], "versa": 281, "version": [2, 4, 9, 88, 126, 150, 152, 190, 227, 232, 258, 300, 331, 514, 520, 521], "versu": 517, "via": [9, 122, 500, 503, 518, 522, 523, 524], "vice": 281, "video": 371, "view": [0, 3, 87, 521, 524], "visual": 151, "vjp": [2, 122, 525], "vmap": [2, 122, 519, 520, 523, 525], "vmap_add": 520, "vocab_s": 6, "vocabulari": [373, 414], "void": [1, 2], "volta": 9, "vt": 219, "w": [0, 1, 5, 108, 109, 112, 113, 126, 180, 207, 208, 258, 259, 327, 341, 361, 364, 365, 367, 368, 370, 371, 385, 490, 505, 520], "w1": [6, 338], "w2": [6, 338], "w3": 6, "w_": [376, 382, 417, 491, 492, 493, 494, 495, 496, 497, 504, 505], "w_i": [126, 258], "w_in": 1, "w_q": 258, "w_star": 5, "w_stride": 1, "w_t": [491, 493, 494, 495, 496, 497, 504, 505], "wa": [4, 6, 87, 135, 136, 518, 519, 523], "wai": [2, 6, 9, 356, 435, 517, 518, 519, 520, 521, 522], "wait": 6, "walk": [6, 519], "walkthrough": 2, "walsh": 187, "want": [1, 2, 6, 518, 519, 520, 522, 527], "warm": [2, 517], "warmup_init": 492, "watch": [6, 517], "wd": 497, "we": [0, 1, 2, 5, 6, 7, 122, 126, 135, 136, 180, 258, 259, 356, 373, 414, 423, 495, 497, 514, 516, 517, 518, 519, 520, 522, 523, 527], "weight": [0, 5, 107, 108, 109, 110, 111, 112, 113, 155, 157, 340, 356, 399, 403, 414, 415, 456, 458, 488, 492, 495, 497, 498, 499, 501, 505, 520, 523], "weight_decai": [492, 495, 497, 499, 505], "weight_fil": 6, "weights_fp16": 523, "well": [6, 356, 397, 409, 412, 518, 523], "wen": 6, "went": 6, "were": [6, 527], "wet": 6, "wget": 9, "what": [2, 6, 340, 522], "whatsoev": 6, "whc": 370, "when": [0, 1, 2, 6, 8, 9, 102, 110, 122, 137, 204, 205, 207, 208, 209, 210, 211, 214, 215, 219, 220, 222, 292, 363, 364, 365, 366, 367, 368, 435, 439, 440, 456, 462, 468, 488, 490, 508, 514, 517, 518, 519, 527], "where": [0, 4, 7, 153, 158, 192, 205, 258, 327, 331, 361, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 374, 376, 377, 381, 382, 383, 385, 396, 413, 416, 417, 432, 439, 440, 445, 446, 448, 459, 465, 471, 475, 477, 482, 501, 518, 520, 521], "wherea": 520, "whether": [150, 152, 154, 156, 180, 208, 210, 218, 220, 259, 376, 382, 396, 412, 417, 456, 459, 465], "which": [0, 1, 2, 6, 7, 8, 9, 19, 38, 87, 88, 102, 110, 128, 131, 132, 133, 135, 136, 137, 146, 150, 152, 158, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 175, 184, 190, 193, 194, 195, 196, 197, 199, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 222, 237, 240, 258, 262, 263, 277, 279, 281, 284, 285, 286, 287, 288, 305, 306, 314, 321, 326, 327, 330, 331, 335, 336, 359, 360, 370, 371, 374, 389, 390, 392, 396, 421, 456, 458, 461, 465, 468, 485, 500, 501, 514, 517, 518, 519, 520, 521, 522, 523, 527, 528], "while": [2, 3, 6, 9, 167, 279, 421, 523, 524], "whistl": 2, "who": 6, "whose": [153, 336, 337], "why": 6, "wi": 518, "wide": 523, "width": [359, 360, 361, 364, 365, 367, 368, 370, 371, 389, 390, 414, 415], "window": [9, 358, 359, 360, 388, 389, 390], "wipe": 9, "wire": 293, "wired_limit_mb": 293, "wise": [0, 2, 13, 14, 20, 21, 22, 23, 24, 25, 26, 92, 93, 94, 95, 99, 115, 116, 138, 139, 143, 144, 145, 147, 149, 176, 177, 185, 186, 192, 201, 202, 203, 223, 224, 225, 226, 227, 229, 230, 231, 235, 243, 245, 247, 249, 255, 275, 276, 280, 283, 294, 295, 296, 297, 303, 304, 310, 316, 317, 362, 370, 371, 380, 391, 413, 425, 444, 451, 452, 454, 455, 470, 471, 473, 474, 477, 478, 479, 480, 517], "wish": 9, "with_logit": 456, "within": [0, 3, 29, 192], "without": [1, 6, 8, 308, 412, 484, 516, 517, 518, 519, 522, 523, 524, 527], "wk": 6, "wl": 2, "wo": 6, "word": 0, "work": [2, 3, 6, 292, 347, 499, 517, 518, 519, 520, 521, 522, 523], "workhors": 356, "world": [343, 518], "world2": 518, "world_ani": 518, "world_mpi": 518, "world_r": 518, "worri": [1, 523], "would": [2, 6, 435, 518, 519, 521, 523, 524, 527], "wq": 6, "wrap": [122, 356, 498], "wrapper": [519, 522], "write": [0, 1, 6, 356, 524], "written": [2, 519], "wrong": 519, "wrt": 337, "wv": 6, "x": [0, 1, 2, 4, 5, 6, 7, 39, 88, 96, 122, 131, 132, 136, 137, 144, 149, 150, 151, 155, 156, 157, 180, 187, 188, 190, 214, 259, 263, 267, 268, 282, 287, 294, 324, 325, 332, 340, 342, 356, 358, 359, 360, 361, 362, 372, 374, 375, 377, 381, 383, 384, 385, 388, 389, 390, 391, 392, 413, 416, 418, 425, 426, 432, 435, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 468, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 488, 490, 497, 517, 518, 519, 520, 521, 523, 524, 525, 527], "x1": 457, "x2": 457, "x86_64": 9, "x_1": [457, 465], "x_2": [457, 465], "x_cast": 2, "x_grad": 1, "x_i": [455, 478, 479], "x_j": [478, 479], "x_offset": 2, "x_ptr": 2, "x_shape": 1, "x_stride": 2, "x_t": [376, 382, 417], "x_view": 524, "xcode": 9, "xcodeproj": 3, "xcrun": 9, "xf": 382, "xg": 382, "xi": 382, "xn": 376, "xo": 382, "xor": 95, "xr": 376, "xy": [0, 237], "xz": 376, "x\u00b2": 524, "y": [0, 2, 4, 5, 6, 7, 9, 39, 88, 122, 150, 151, 187, 190, 332, 356, 361, 370, 377, 381, 383, 385, 416, 460, 465, 468, 490, 493, 517, 518, 519, 520, 523, 524], "y_": [460, 464], "y_cast": 2, "y_hat": 356, "y_offset": 2, "y_ptr": 2, "y_stride": 2, "ye": 6, "year": 6, "yet": [6, 356, 488, 501, 520, 521, 523, 525], "yield": [6, 7, 514], "you": [1, 2, 3, 4, 6, 7, 8, 9, 293, 356, 426, 434, 485, 514, 517, 518, 519, 520, 521, 522, 524, 526, 527], "your": [2, 6, 9, 488, 518, 520, 523], "z": [2, 88, 376, 517, 519, 523], "z_t": 376, "zeiler": 491, "zero": [0, 150, 153, 160, 161, 162, 163, 164, 165, 166, 168, 169, 170, 171, 172, 173, 206, 237, 278, 299, 323, 324, 325, 334, 356, 358, 359, 360, 369, 370, 371, 399, 436, 437, 438, 439, 440, 441, 442, 443, 485, 490, 492, 519, 521], "zero_grad": 520, "zeros_lik": [0, 212], "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.imag", "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.logcumsumexp", "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.real", "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.cuda.is_available", "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.cuda_kernel", "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.fftshift", "mlx.core.fft.ifft", "mlx.core.fft.ifft2", "mlx.core.fft.ifftn", "mlx.core.fft.ifftshift", "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.eig", "mlx.core.linalg.eigh", "mlx.core.linalg.eigvals", "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.logcumsumexp", "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", "CUDA", "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.ReLU2", "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.relu2", "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.MultiOptimizer", "mlx.optimizers.Muon", "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": 527, "In": 521, "The": 356, "ab": [13, 33], "adadelta": 491, "adafactor": 492, "adagrad": 493, "adam": 494, "adamax": 496, "adamw": 495, "add": 14, "addmm": 15, "algebra": 353, "alibi": 357, "all": [6, 16, 34, 518], "all_gath": 131, "all_sum": 132, "allclos": 17, "ani": [18, 35], "api": [8, 9], "appli": 392, "apply_gradi": 500, "apply_to_modul": 393, "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, 83, 84, 85, 345, 521, 526], "array_equ": 86, "as_strid": 87, "astyp": 38, "async_ev": 88, "atleast_1d": 89, "atleast_2d": 90, "atleast_3d": 91, "attent": 6, "automat": 520, "average_gradi": [335, 518], "avgpool1d": 358, "avgpool2d": 359, "avgpool3d": 360, "back": 2, "backend": 518, "basic": [517, 519, 525], "batchnorm": 361, "benchmark": 6, "bernoulli": 261, "binari": 9, "binary_cross_entropi": 456, "bind": 2, "bitwise_and": 92, "bitwise_invert": 93, "bitwise_or": 94, "bitwise_xor": 95, "block_masked_mm": 96, "broadcast_arrai": 97, "broadcast_to": 98, "build": [2, 9], "c": [4, 8, 9, 519], "categor": 262, "ceil": 99, "celu": [362, 444], "children": 394, "choleski": 204, "cholesky_inv": 205, "class": 356, "clear_cach": 100, "clip": 101, "clip_grad_norm": 338, "cmake": 2, "co": [41, 115], "code": [2, 6], "common": 511, "commun": [349, 518], "compil": [102, 517], "complex": 1, "comput": 523, "concaten": 103, "conj": [40, 104], "conjug": 105, "constant": 436, "contigu": 106, "conv1d": [107, 363], "conv2d": [108, 364], "conv3d": [109, 365], "conv_gener": 110, "conv_transpose1d": 111, "conv_transpose2d": 112, "conv_transpose3d": 113, "convers": 524, "convert": 6, "convolv": 114, "convtranspose1d": 366, "convtranspose2d": 367, "convtranspose3d": 368, "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, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 344], "cosh": 116, "cosine_decai": 506, "cosine_similarity_loss": 457, "cpu": [2, 9], "cross": 206, "cross_entropi": 458, "cuda": [9, 117, 346], "cuda_kernel": 154, "cummax": [42, 118], "cummin": [43, 119], "cumprod": [44, 120], "cumsum": [45, 121], "custom": [1, 2], "custom_funct": 122, "data": 347, "debug": 517, "debugg": 3, "default_devic": 123, "default_stream": 124, "defin": 518, "degre": 125, "dequant": 126, "devic": [10, 348], "device_info": 238, "diag": [46, 127], "diagon": [47, 128], "differ": 521, "differenti": 520, "disable_compil": 129, "distribut": [130, 131, 132, 133, 134, 135, 136, 137, 349, 518, 522], "divid": 138, "divmod": 139, "download": [2, 6], "dropout": 369, "dropout2d": 370, "dropout3d": 371, "dtype": [11, 48], "dtypecategori": 12, "eig": 207, "eigh": 208, "eigval": 209, "eigvalsh": 210, "einsum": 140, "einsum_path": 141, "elu": [372, 445], "embed": 373, "enable_compil": 142, "encod": 6, "end": 2, "equal": 143, "erf": 144, "erfinv": 145, "eval": [146, 395], "evalu": 523, "exampl": [1, 2, 8, 517, 518, 519, 527], "exp": [49, 147], "expand_dim": 148, "expm1": 149, "exponential_decai": 507, "export": [152, 350, 519], "export_funct": 150, "export_to_dot": 151, "extens": 2, "ey": 153, "fast": [154, 155, 156, 157, 158, 159, 351], "fft": [160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 352], "fft2": 161, "fftn": 162, "fftshift": 163, "filter_and_map": 396, "finfo": 174, "flatten": [50, 175], "floor": 176, "floor_divid": 177, "format": 526, "found": 9, "framework": 524, "freez": 397, "from": [9, 521], "full": [6, 178], "function": [350, 484, 487, 517, 519, 520, 525], "further": 8, "gather_mm": 179, "gather_qmm": 180, "gaussian_nll_loss": 459, "gelu": [374, 446], "gelu_approx": 447, "gelu_fast_approx": 448, "gener": 6, "get": 518, "get_active_memori": 181, "get_cache_memori": 182, "get_peak_memori": 183, "glorot_norm": 437, "glorot_uniform": 438, "glu": [375, 449], "gpu": 2, "grad": [184, 356], "graph": [517, 523, 525], "greater": 185, "greater_equ": 186, "grid": 1, "group": 130, "groupnorm": 377, "gru": 376, "guid": 525, "gumbel": 263, "hadamard_transform": 187, "hard_shrink": 450, "hard_tanh": 451, "hardshrink": 378, "hardswish": [380, 452], "hardtanh": 379, "he_norm": 439, "he_uniform": 440, "hinge_loss": 460, "host": [518, 522], "huber_loss": 461, "ident": [188, 441], "ifft": 164, "ifft2": 165, "ifftn": 166, "ifftshift": 167, "imag": [51, 189], "implement": [2, 6], "import": 519, "import_funct": 190, "index": 521, "infer": 6, "init": [133, 436, 437, 438, 439, 440, 441, 442, 443, 501], "initi": 485, "inner": 191, "inspect": 356, "instal": [8, 9, 518], "instancenorm": 381, "introduc": 2, "inv": 211, "irfft": 168, "irfft2": 169, "irfftn": 170, "is_avail": [117, 134, 239], "isclos": 192, "isfinit": 193, "isinf": 194, "isnan": 195, "isneginf": 196, "isposinf": 197, "issubdtyp": 198, "item": 52, "items": 53, "jax": 524, "join_schedul": 508, "jvp": 199, "kei": 264, "kernel": 1, "kl_div_loss": 462, "kron": 200, "l1_loss": 463, "laplac": 265, "launch": 522, "layer": [6, 7, 486], "layer_norm": 155, "layernorm": 383, "lazi": 523, "leaf_modul": 398, "leaky_relu": 453, "leakyrelu": 384, "left_shift": 201, "less": 202, "less_equ": 203, "linalg": [204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220], "linear": [5, 353, 385], "linear_schedul": 509, "linspac": 221, "linux": 9, "lion": 497, "llm": 6, "load": [6, 222, 490, 526], "load_weight": 399, "log": [54, 223], "log10": [55, 224], "log1p": [56, 225], "log2": [57, 226], "log_cosh_loss": 464, "log_sigmoid": 454, "log_softmax": 455, "logaddexp": 227, "logcumsumexp": [58, 228], "logical_and": 229, "logical_not": 230, "logical_or": 231, "logsigmoid": 386, "logsoftmax": 387, "logsumexp": [59, 232], "loss": [456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 487], "lstm": 382, "lu": 212, "lu_factor": 213, "manag": 354, "margin_ranking_loss": 465, "matmul": 233, "max": [60, 234], "maximum": 235, "maxpool1d": 388, "maxpool2d": 389, "maxpool3d": 390, "mean": [61, 236], "memori": [354, 527], "meshgrid": 237, "metal": [1, 3, 9, 238, 239, 240, 241, 355], "metal_kernel": 156, "min": [62, 242], "minim": 9, "minimum": 243, "mish": [391, 470], "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, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 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, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510], "model": 6, "modul": [356, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 488, 519], "more": 519, "moveaxi": [63, 244], "mpi": [518, 522], "mse_loss": 466, "multi": 7, "multiheadattent": 412, "multioptim": 498, "multipl": 519, "multipli": 245, "multivariate_norm": 266, "muon": 499, "named_modul": 401, "nan_to_num": 246, "nbyte": 64, "ndim": 65, "neg": 247, "network": 356, "neural": 356, "new_stream": 248, "nll_loss": 467, "nn": [335, 336, 337, 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, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 518], "norm": 214, "normal": [267, 442], "not_equ": 249, "numpi": [521, 524], "ones": 250, "ones_lik": 251, "onli": [9, 523], "oper": [0, 2, 489], "optim": [338, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512], "option": 9, "other": 524, "outer": 252, "packag": 4, "pad": 253, "paramet": [356, 402], "partit": 254, "perceptron": 7, "permut": 268, "pinv": 215, "place": 521, "power": 255, "prelu": [413, 471], "primit": 2, "prod": [66, 256], "program": [518, 522], "provid": 522, "pure": 517, "put": 6, "put_along_axi": 257, "python": [2, 8, 9], "pytorch": 524, "qr": 216, "quantiz": [258, 336], "quantized_matmul": 259, "quantizedembed": 414, "quantizedlinear": 415, "quick": [356, 525], "radian": 260, "randint": 269, "random": [261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 514], "read": 8, "real": [67, 274], "reciproc": [68, 275], "recv": 135, "recv_lik": 136, "reduc": 518, "refer": 8, "regress": 5, "relu": [418, 472], "relu2": [419, 473], "relu6": [420, 474], "remaind": 276, "remot": [518, 522], "repeat": 277, "requir": 9, "reset_peak_memori": 278, "reshap": [69, 279], "result": 2, "rfft": 171, "rfft2": 172, "rfftn": 173, "right_shift": 280, "ring": [518, 522], "rms_norm": 157, "rmsnorm": 416, "rmsprop": 504, "rnn": 417, "roll": 281, "rope": [158, 421], "round": [70, 282], "rsqrt": [71, 283], "run": 518, "sampl": 1, "save": [284, 490, 526], "save_gguf": 285, "save_safetensor": 286, "save_weight": 403, "savez": 287, "savez_compress": 288, "scaled_dot_product_attent": 159, "schedul": 513, "script": [2, 6], "seed": 270, "select": 518, "selu": [422, 475], "send": 137, "sequenti": 423, "serial": 526, "set": [518, 522], "set_cache_limit": 289, "set_default_devic": 290, "set_default_stream": 291, "set_dtyp": 404, "set_memory_limit": 292, "set_wired_limit": 293, "setuptool": 2, "sgd": 505, "shape": [1, 72], "shapeless": [517, 519], "shell": 9, "sigmoid": [294, 425, 476], "sign": 295, "silu": [424, 477], "simpl": [1, 527], "sin": [73, 296], "sinh": 297, "sinusoidalpositionalencod": 426, "size": [9, 74], "slice": 298, "slice_upd": 299, "smooth_l1_loss": 468, "softmax": [300, 427, 478], "softmin": [428, 479], "softplu": [429, 480], "softshrink": [430, 481], "softsign": 431, "solv": 217, "solve_triangular": 218, "sort": 301, "sourc": 9, "specif": 522, "specifi": 528, "speedup": 517, "split": [75, 271, 302], "sqrt": [76, 303], "squar": [77, 304], "squeez": [78, 305], "stack": 306, "start": [356, 518, 525], "start_captur": 240, "state": [405, 502], "std": [79, 307], "step": [432, 482], "step_decai": 510, "stop_captur": 241, "stop_gradi": 308, "stream": [309, 344, 348, 528], "stride": 1, "subtract": 310, "sum": [80, 311], "support": 347, "svd": 219, "swapax": [81, 312], "synchron": 313, "t": 32, "take": 314, "take_along_axi": 315, "tan": 316, "tanh": [317, 433, 483], "tensordot": 318, "tensorflow": 524, "thunderbolt": 518, "tile": 319, "togeth": 6, "tolist": 82, "topk": 320, "trace": [321, 519], "train": [406, 408, 517, 518], "trainable_paramet": 407, "transform": [2, 434, 515, 517, 519, 520, 523, 525], "transpos": [83, 322], "tree": 516, "tree_flatten": 339, "tree_map": 340, "tree_map_with_path": 341, "tree_reduc": 342, "tree_unflatten": 343, "tri": 323, "tri_inv": 220, "tril": 324, "triplet_loss": 469, "triu": 325, "troubleshoot": 9, "truncated_norm": 272, "tune": 518, "type": 347, "unflatten": 326, "unfreez": 409, "unifi": 527, "uniform": [273, 443], "up": [518, 522], "updat": [356, 410, 503, 521], "update_modul": 411, "upsampl": 435, "us": [1, 2, 4, 523, 528], "usag": [2, 8, 522], "util": [339, 340, 341, 342, 343, 516, 518], "valu": 356, "value_and_grad": [327, 337], "var": [84, 328], "variabl": 4, "vector": 520, "view": [85, 329], "vjp": [1, 330], "vmap": 331, "weight": 6, "what": 523, "when": 523, "where": 332, "why": 523, "workflow": 3, "x86": 9, "xcode": 3, "you": 523, "zero": 333, "zeros_lik": 334}}) \ No newline at end of file diff --git a/docs/build/html/usage/compile.html b/docs/build/html/usage/compile.html index 6275ebd1d..830fe8645 100644 --- a/docs/build/html/usage/compile.html +++ b/docs/build/html/usage/compile.html @@ -8,7 +8,7 @@ - Compilation — MLX 0.29.2 documentation + Compilation — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/usage/distributed.html b/docs/build/html/usage/distributed.html index ef2cd27a2..2315a6d53 100644 --- a/docs/build/html/usage/distributed.html +++ b/docs/build/html/usage/distributed.html @@ -8,7 +8,7 @@ - Distributed Communication — MLX 0.29.2 documentation + Distributed Communication — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/usage/export.html b/docs/build/html/usage/export.html index 57e4bf484..a11668f6d 100644 --- a/docs/build/html/usage/export.html +++ b/docs/build/html/usage/export.html @@ -8,7 +8,7 @@ - Exporting Functions — MLX 0.29.2 documentation + Exporting Functions — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + diff --git a/docs/build/html/usage/function_transforms.html b/docs/build/html/usage/function_transforms.html index 9e9cb5ea2..17e15a558 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.29.2 documentation + Function Transforms — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -1009,8 +1009,8 @@ should use list, tuple, or -dict).

                                      +containers of arrays (specifically any of list, tuple, or +dict).

                                      Suppose we wanted a weight and a bias parameter in the above example. A nice way to do that is the following:

                                      def loss_fn(params, x, y):
diff --git a/docs/build/html/usage/indexing.html b/docs/build/html/usage/indexing.html
index 2250b063d..bd3521417 100644
--- a/docs/build/html/usage/indexing.html
+++ b/docs/build/html/usage/indexing.html
@@ -8,7 +8,7 @@
     
     
 
-    Indexing Arrays — MLX 0.29.2 documentation
+    Indexing Arrays — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                                      @@ -944,7 +944,7 @@ array(8, ([2, 4, 6], dtype=int32)
                                      -

                                      For multi-dimensional arrays, the ... or Ellipsis syntax works as in NumPy:

                                      +

                                      For multi-dimensional arrays, the ... or Ellipsis syntax works as in NumPy:

                                      >>> arr = mx.arange(8).reshape(2, 2, 2)
 >>> arr[:, :, 0]
 array(3, dtype=int32)
diff --git a/docs/build/html/usage/launching_distributed.html b/docs/build/html/usage/launching_distributed.html
index efa8c516f..617c84938 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.29.2 documentation
+    Launching Distributed Programs — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                                      diff --git a/docs/build/html/usage/lazy_evaluation.html b/docs/build/html/usage/lazy_evaluation.html index 5c864bdef..047743a9e 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.29.2 documentation + Lazy Evaluation — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +
                                      @@ -1019,7 +1019,7 @@ stochastic gradient descent). A natural and usually efficient place to use

                                      An important behavior to be aware of is when the graph will be implicitly evaluated. Anytime you print an array, convert it to an -numpy.ndarray, or otherwise access its memory via memoryview, +numpy.ndarray, or otherwise access its memory via memoryview, the graph will be evaluated. Saving arrays via save() (or any other MLX saving functions) will also evaluate the array.

                                      Calling array.item() on a scalar array will also evaluate it. In the diff --git a/docs/build/html/usage/numpy.html b/docs/build/html/usage/numpy.html index 125b37ec3..36b0ffe76 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.29.2 documentation + Conversion to NumPy and Other Frameworks — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home + @@ -995,11 +995,11 @@ even though no in-place operations on MLX memory are executed.

                                      PyTorch#

                                      Warning

                                      -

                                      PyTorch Support for memoryview is experimental and can break for +

                                      PyTorch Support for memoryview is experimental and can break for multi-dimensional arrays. Casting to NumPy first is advised for now.

                                      PyTorch supports the buffer protocol, but it requires an explicit -memoryview.

                                      +memoryview.

                                      import mlx.core as mx
 import torch
 
@@ -1026,7 +1026,7 @@ NumPy arrays with n
 
                                      
 
                                      TensorFlow#
                                      
 
                                      TensorFlow supports the buffer protocol, but it requires an explicit
-memoryview.
                                      
+memoryview.
                                      
 
                                      import mlx.core as mx
 import tensorflow as tf
 
diff --git a/docs/build/html/usage/quick_start.html b/docs/build/html/usage/quick_start.html
index e4d6e82b8..a3055f655 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.29.2 documentation
+    Quick Start Guide — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+    
   
   
 
                                      
diff --git a/docs/build/html/usage/saving_and_loading.html b/docs/build/html/usage/saving_and_loading.html
index d60c1488a..b77103939 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.29.2 documentation
+    Saving and Loading Arrays — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+    
   
   
 
                                      
@@ -990,7 +990,7 @@ provided. This can be loaded with:

                                      In this case load() returns a dictionary of names to arrays.

                                      The functions save_safetensors() and save_gguf() are similar to -savez(), but they take as input a dict of string names to arrays:

                                      +savez(), but they take as input a dict of string names to arrays:

                                      >>> a = mx.array([1.0])
 >>> b = mx.array([2.0])
 >>> mx.save_safetensors("arrays", {"a": a, "b": b})
diff --git a/docs/build/html/usage/unified_memory.html b/docs/build/html/usage/unified_memory.html
index ef5bb6d01..c88b96fce 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.29.2 documentation
+    Unified Memory — MLX 0.29.3 documentation
   
   
   
@@ -37,7 +37,7 @@
 
   
 
-    
+    
     
     
     
@@ -140,8 +140,8 @@
       
     
     
-    MLX 0.29.2 documentation - Home
-    
+    MLX 0.29.3 documentation - Home
+
                                      diff --git a/docs/build/html/usage/using_streams.html b/docs/build/html/usage/using_streams.html index fbaadddc3..5b4f8c38c 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.29.2 documentation + Using Streams — MLX 0.29.3 documentation @@ -37,7 +37,7 @@ - + @@ -140,8 +140,8 @@ - MLX 0.29.2 documentation - Home - + MLX 0.29.3 documentation - Home +

                          export_function(file, fun, *args[, shapeless])

                          Export a function to a file.

                          export_function(arg0, fun, *args[, shapeless])

                          Export an MLX function.

                          import_function(file)

                          Import a function from a file.