mlx/_autosummary at 1e46b74a18184839b166787fe14563b7e6f4ad22 - 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.logcumsumexp.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.async_eval.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_arrays.rst

mlx.core.broadcast_to.rst

mlx.core.ceil.rst

mlx.core.clear_cache.rst

mlx.core.clip.rst

mlx.core.compile.rst

mlx.core.concatenate.rst

mlx.core.conj.rst

mlx.core.conjugate.rst

mlx.core.contiguous.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.get_active_memory.rst

mlx.core.get_cache_memory.rst

mlx.core.get_peak_memory.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.pinv.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.logcumsumexp.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.device_info.rst

mlx.core.metal.is_available.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.reset_peak_memory.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_cache_limit.rst

mlx.core.set_default_device.rst

mlx.core.set_default_stream.rst

mlx.core.set_memory_limit.rst

mlx.core.set_wired_limit.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.average_gradients.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