mlx/_autosummary at cc4f6f1107f1bef53f8723be6ac48ced48f689d6 - 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