mlx/_autosummary at 8a2a679c9cfd14d4dcd7d3ffb0c1204bf30ea484 - mlx

mlx.core.abs.rst

mlx.core.add.rst

mlx.core.addmm.rst

mlx.core.all.rst

mlx.core.allclose.rst

mlx.core.any.rst

mlx.core.arange.rst

mlx.core.arccos.rst

mlx.core.arccosh.rst

mlx.core.arcsin.rst

mlx.core.arcsinh.rst

mlx.core.arctan2.rst

mlx.core.arctan.rst

mlx.core.arctanh.rst

mlx.core.argmax.rst

mlx.core.argmin.rst

mlx.core.argpartition.rst

mlx.core.argsort.rst

mlx.core.array_equal.rst

mlx.core.array.abs.rst

mlx.core.array.all.rst

mlx.core.array.any.rst

mlx.core.array.argmax.rst

mlx.core.array.argmin.rst

mlx.core.array.astype.rst

mlx.core.array.at.rst

mlx.core.array.conj.rst

mlx.core.array.cos.rst

mlx.core.array.cummax.rst

mlx.core.array.cummin.rst

mlx.core.array.cumprod.rst

mlx.core.array.cumsum.rst

mlx.core.array.diag.rst

mlx.core.array.diagonal.rst

mlx.core.array.dtype.rst

mlx.core.array.exp.rst

mlx.core.array.flatten.rst

mlx.core.array.item.rst

mlx.core.array.itemsize.rst

mlx.core.array.log1p.rst

mlx.core.array.log2.rst

mlx.core.array.log10.rst

mlx.core.array.log.rst

mlx.core.array.logsumexp.rst

mlx.core.array.max.rst

mlx.core.array.mean.rst

mlx.core.array.min.rst

mlx.core.array.moveaxis.rst

mlx.core.array.nbytes.rst

mlx.core.array.ndim.rst

mlx.core.array.prod.rst

mlx.core.array.reciprocal.rst

mlx.core.array.reshape.rst

mlx.core.array.round.rst

mlx.core.array.rsqrt.rst

mlx.core.array.rst

mlx.core.array.shape.rst

mlx.core.array.sin.rst

mlx.core.array.size.rst

mlx.core.array.split.rst

mlx.core.array.sqrt.rst

mlx.core.array.square.rst

mlx.core.array.squeeze.rst

mlx.core.array.std.rst

mlx.core.array.sum.rst

mlx.core.array.swapaxes.rst

mlx.core.array.T.rst

mlx.core.array.tolist.rst

mlx.core.array.transpose.rst

mlx.core.array.var.rst

mlx.core.array.view.rst

mlx.core.as_strided.rst

mlx.core.atleast_1d.rst

mlx.core.atleast_2d.rst

mlx.core.atleast_3d.rst

mlx.core.bitwise_and.rst

mlx.core.bitwise_invert.rst

mlx.core.bitwise_or.rst

mlx.core.bitwise_xor.rst

mlx.core.block_masked_mm.rst

mlx.core.broadcast_to.rst

mlx.core.ceil.rst

mlx.core.clip.rst

mlx.core.compile.rst

mlx.core.concatenate.rst

mlx.core.conj.rst

mlx.core.conjugate.rst

mlx.core.conv1d.rst

mlx.core.conv2d.rst

mlx.core.conv3d.rst

mlx.core.conv_general.rst

mlx.core.conv_transpose1d.rst

mlx.core.conv_transpose2d.rst

mlx.core.conv_transpose3d.rst

mlx.core.convolve.rst

mlx.core.cos.rst

mlx.core.cosh.rst

mlx.core.cummax.rst

mlx.core.cummin.rst

mlx.core.cumprod.rst

mlx.core.cumsum.rst

mlx.core.custom_function.rst

mlx.core.default_device.rst

mlx.core.default_stream.rst

mlx.core.degrees.rst

mlx.core.dequantize.rst

mlx.core.Device.rst

mlx.core.diag.rst

mlx.core.diagonal.rst

mlx.core.disable_compile.rst

mlx.core.distributed.all_gather.rst

mlx.core.distributed.all_sum.rst

mlx.core.distributed.Group.rst

mlx.core.distributed.init.rst

mlx.core.distributed.is_available.rst

mlx.core.distributed.recv_like.rst

mlx.core.distributed.recv.rst

mlx.core.distributed.send.rst

mlx.core.divide.rst

mlx.core.divmod.rst

mlx.core.Dtype.rst

mlx.core.DtypeCategory.rst

mlx.core.einsum_path.rst

mlx.core.einsum.rst

mlx.core.enable_compile.rst

mlx.core.equal.rst

mlx.core.erf.rst

mlx.core.erfinv.rst

mlx.core.eval.rst

mlx.core.exp.rst

mlx.core.expand_dims.rst

mlx.core.expm1.rst

mlx.core.export_function.rst

mlx.core.export_to_dot.rst

mlx.core.exporter.rst

mlx.core.eye.rst

mlx.core.fast.layer_norm.rst

mlx.core.fast.metal_kernel.rst

mlx.core.fast.rms_norm.rst

mlx.core.fast.rope.rst

mlx.core.fast.scaled_dot_product_attention.rst

mlx.core.fft.fft2.rst

mlx.core.fft.fft.rst

mlx.core.fft.fftn.rst

mlx.core.fft.ifft2.rst

mlx.core.fft.ifft.rst

mlx.core.fft.ifftn.rst

mlx.core.fft.irfft2.rst

mlx.core.fft.irfft.rst

mlx.core.fft.irfftn.rst

mlx.core.fft.rfft2.rst

mlx.core.fft.rfft.rst

mlx.core.fft.rfftn.rst

mlx.core.finfo.rst

mlx.core.flatten.rst

mlx.core.floor_divide.rst

mlx.core.floor.rst

mlx.core.full.rst

mlx.core.gather_mm.rst

mlx.core.gather_qmm.rst

mlx.core.grad.rst

mlx.core.greater_equal.rst

mlx.core.greater.rst

mlx.core.hadamard_transform.rst

mlx.core.identity.rst

mlx.core.imag.rst

mlx.core.import_function.rst

mlx.core.inner.rst

mlx.core.isclose.rst

mlx.core.isfinite.rst

mlx.core.isinf.rst

mlx.core.isnan.rst

mlx.core.isneginf.rst

mlx.core.isposinf.rst

mlx.core.issubdtype.rst

mlx.core.jvp.rst

mlx.core.kron.rst

mlx.core.left_shift.rst

mlx.core.less_equal.rst

mlx.core.less.rst

mlx.core.linalg.cholesky_inv.rst

mlx.core.linalg.cholesky.rst

mlx.core.linalg.cross.rst

mlx.core.linalg.eigh.rst

mlx.core.linalg.eigvalsh.rst

mlx.core.linalg.inv.rst

mlx.core.linalg.lu_factor.rst

mlx.core.linalg.lu.rst

mlx.core.linalg.norm.rst

mlx.core.linalg.qr.rst

mlx.core.linalg.solve_triangular.rst

mlx.core.linalg.solve.rst

mlx.core.linalg.svd.rst

mlx.core.linalg.tri_inv.rst

mlx.core.linspace.rst

mlx.core.load.rst

mlx.core.log1p.rst

mlx.core.log2.rst

mlx.core.log10.rst

mlx.core.log.rst

mlx.core.logaddexp.rst

mlx.core.logical_and.rst

mlx.core.logical_not.rst

mlx.core.logical_or.rst

mlx.core.logsumexp.rst

mlx.core.matmul.rst

mlx.core.max.rst

mlx.core.maximum.rst

mlx.core.mean.rst

mlx.core.meshgrid.rst

mlx.core.metal.clear_cache.rst

mlx.core.metal.device_info.rst

mlx.core.metal.get_active_memory.rst

mlx.core.metal.get_cache_memory.rst

mlx.core.metal.get_peak_memory.rst

mlx.core.metal.is_available.rst

mlx.core.metal.reset_peak_memory.rst

mlx.core.metal.set_cache_limit.rst

mlx.core.metal.set_memory_limit.rst

mlx.core.metal.set_wired_limit.rst

mlx.core.metal.start_capture.rst

mlx.core.metal.stop_capture.rst

mlx.core.min.rst

mlx.core.minimum.rst

mlx.core.moveaxis.rst

mlx.core.multiply.rst

mlx.core.nan_to_num.rst

mlx.core.negative.rst

mlx.core.new_stream.rst

mlx.core.not_equal.rst

mlx.core.ones_like.rst

mlx.core.ones.rst

mlx.core.outer.rst

mlx.core.pad.rst

mlx.core.partition.rst

mlx.core.power.rst

mlx.core.prod.rst

mlx.core.put_along_axis.rst

mlx.core.quantize.rst

mlx.core.quantized_matmul.rst

mlx.core.radians.rst

mlx.core.random.bernoulli.rst

mlx.core.random.categorical.rst

mlx.core.random.gumbel.rst

mlx.core.random.key.rst

mlx.core.random.laplace.rst

mlx.core.random.multivariate_normal.rst

mlx.core.random.normal.rst

mlx.core.random.permutation.rst

mlx.core.random.randint.rst

mlx.core.random.seed.rst

mlx.core.random.split.rst

mlx.core.random.truncated_normal.rst

mlx.core.random.uniform.rst

mlx.core.real.rst

mlx.core.reciprocal.rst

mlx.core.remainder.rst

mlx.core.repeat.rst

mlx.core.reshape.rst

mlx.core.right_shift.rst

mlx.core.roll.rst

mlx.core.round.rst

mlx.core.rsqrt.rst

mlx.core.save_gguf.rst

mlx.core.save_safetensors.rst

mlx.core.save.rst

mlx.core.savez_compressed.rst

mlx.core.savez.rst

mlx.core.set_default_device.rst

mlx.core.set_default_stream.rst

mlx.core.sigmoid.rst

mlx.core.sign.rst

mlx.core.sin.rst

mlx.core.sinh.rst

mlx.core.slice_update.rst

mlx.core.slice.rst

mlx.core.softmax.rst

mlx.core.sort.rst

mlx.core.split.rst

mlx.core.sqrt.rst

mlx.core.square.rst

mlx.core.squeeze.rst

mlx.core.stack.rst

mlx.core.std.rst

mlx.core.stop_gradient.rst

mlx.core.stream.rst

mlx.core.subtract.rst

mlx.core.sum.rst

mlx.core.swapaxes.rst

mlx.core.synchronize.rst

mlx.core.take_along_axis.rst

mlx.core.take.rst

mlx.core.tan.rst

mlx.core.tanh.rst

mlx.core.tensordot.rst

mlx.core.tile.rst

mlx.core.topk.rst

mlx.core.trace.rst

mlx.core.transpose.rst

mlx.core.tri.rst

mlx.core.tril.rst

mlx.core.triu.rst

mlx.core.unflatten.rst

mlx.core.value_and_grad.rst

mlx.core.var.rst

mlx.core.view.rst

mlx.core.vjp.rst

mlx.core.vmap.rst

mlx.core.where.rst

mlx.core.zeros_like.rst

mlx.core.zeros.rst

mlx.nn.value_and_grad.rst

mlx.optimizers.clip_grad_norm.rst

mlx.utils.tree_flatten.rst

mlx.utils.tree_map_with_path.rst

mlx.utils.tree_map.rst

mlx.utils.tree_reduce.rst

mlx.utils.tree_unflatten.rst

stream_class.rst