| .. |
|
mlx.core.abs.html
|
|
|
|
mlx.core.add.html
|
|
|
|
mlx.core.addmm.html
|
|
|
|
mlx.core.all.html
|
|
|
|
mlx.core.allclose.html
|
|
|
|
mlx.core.any.html
|
|
|
|
mlx.core.arange.html
|
|
|
|
mlx.core.arccos.html
|
|
|
|
mlx.core.arccosh.html
|
|
|
|
mlx.core.arcsin.html
|
|
|
|
mlx.core.arcsinh.html
|
|
|
|
mlx.core.arctan2.html
|
|
|
|
mlx.core.arctan.html
|
|
|
|
mlx.core.arctanh.html
|
|
|
|
mlx.core.argmax.html
|
|
|
|
mlx.core.argmin.html
|
|
|
|
mlx.core.argpartition.html
|
|
|
|
mlx.core.argsort.html
|
|
|
|
mlx.core.array_equal.html
|
|
|
|
mlx.core.array.abs.html
|
|
|
|
mlx.core.array.all.html
|
|
|
|
mlx.core.array.any.html
|
|
|
|
mlx.core.array.argmax.html
|
|
|
|
mlx.core.array.argmin.html
|
|
|
|
mlx.core.array.astype.html
|
|
|
|
mlx.core.array.at.html
|
|
|
|
mlx.core.array.conj.html
|
|
|
|
mlx.core.array.cos.html
|
|
|
|
mlx.core.array.cummax.html
|
|
|
|
mlx.core.array.cummin.html
|
|
|
|
mlx.core.array.cumprod.html
|
|
|
|
mlx.core.array.cumsum.html
|
|
|
|
mlx.core.array.diag.html
|
|
|
|
mlx.core.array.diagonal.html
|
|
|
|
mlx.core.array.dtype.html
|
|
|
|
mlx.core.array.exp.html
|
|
|
|
mlx.core.array.flatten.html
|
|
|
|
mlx.core.array.html
|
|
|
|
mlx.core.array.imag.html
|
|
|
|
mlx.core.array.item.html
|
|
|
|
mlx.core.array.itemsize.html
|
|
|
|
mlx.core.array.log1p.html
|
|
|
|
mlx.core.array.log2.html
|
|
|
|
mlx.core.array.log10.html
|
|
|
|
mlx.core.array.log.html
|
|
|
|
mlx.core.array.logcumsumexp.html
|
|
|
|
mlx.core.array.logsumexp.html
|
|
|
|
mlx.core.array.max.html
|
|
|
|
mlx.core.array.mean.html
|
|
|
|
mlx.core.array.min.html
|
|
|
|
mlx.core.array.moveaxis.html
|
|
|
|
mlx.core.array.nbytes.html
|
|
|
|
mlx.core.array.ndim.html
|
|
|
|
mlx.core.array.prod.html
|
|
|
|
mlx.core.array.real.html
|
|
|
|
mlx.core.array.reciprocal.html
|
|
|
|
mlx.core.array.reshape.html
|
|
|
|
mlx.core.array.round.html
|
|
|
|
mlx.core.array.rsqrt.html
|
|
|
|
mlx.core.array.shape.html
|
|
|
|
mlx.core.array.sin.html
|
|
|
|
mlx.core.array.size.html
|
|
|
|
mlx.core.array.split.html
|
|
|
|
mlx.core.array.sqrt.html
|
|
|
|
mlx.core.array.square.html
|
|
|
|
mlx.core.array.squeeze.html
|
|
|
|
mlx.core.array.std.html
|
|
|
|
mlx.core.array.sum.html
|
|
|
|
mlx.core.array.swapaxes.html
|
|
|
|
mlx.core.array.T.html
|
|
|
|
mlx.core.array.tolist.html
|
|
|
|
mlx.core.array.transpose.html
|
|
|
|
mlx.core.array.var.html
|
|
|
|
mlx.core.array.view.html
|
|
|
|
mlx.core.as_strided.html
|
|
|
|
mlx.core.async_eval.html
|
|
|
|
mlx.core.atleast_1d.html
|
|
|
|
mlx.core.atleast_2d.html
|
|
|
|
mlx.core.atleast_3d.html
|
|
|
|
mlx.core.bitwise_and.html
|
|
|
|
mlx.core.bitwise_invert.html
|
|
|
|
mlx.core.bitwise_or.html
|
|
|
|
mlx.core.bitwise_xor.html
|
|
|
|
mlx.core.block_masked_mm.html
|
|
|
|
mlx.core.broadcast_arrays.html
|
|
|
|
mlx.core.broadcast_to.html
|
|
|
|
mlx.core.ceil.html
|
|
|
|
mlx.core.clear_cache.html
|
|
|
|
mlx.core.clip.html
|
|
|
|
mlx.core.compile.html
|
|
|
|
mlx.core.concatenate.html
|
|
|
|
mlx.core.conj.html
|
|
|
|
mlx.core.conjugate.html
|
|
|
|
mlx.core.contiguous.html
|
|
|
|
mlx.core.conv1d.html
|
|
|
|
mlx.core.conv2d.html
|
|
|
|
mlx.core.conv3d.html
|
|
|
|
mlx.core.conv_general.html
|
|
|
|
mlx.core.conv_transpose1d.html
|
|
|
|
mlx.core.conv_transpose2d.html
|
|
|
|
mlx.core.conv_transpose3d.html
|
|
|
|
mlx.core.convolve.html
|
|
|
|
mlx.core.cos.html
|
|
|
|
mlx.core.cosh.html
|
|
|
|
mlx.core.cummax.html
|
|
|
|
mlx.core.cummin.html
|
|
|
|
mlx.core.cumprod.html
|
|
|
|
mlx.core.cumsum.html
|
|
|
|
mlx.core.custom_function.html
|
|
|
|
mlx.core.default_device.html
|
|
|
|
mlx.core.default_stream.html
|
|
|
|
mlx.core.degrees.html
|
|
|
|
mlx.core.dequantize.html
|
|
|
|
mlx.core.Device.html
|
|
|
|
mlx.core.diag.html
|
|
|
|
mlx.core.diagonal.html
|
|
|
|
mlx.core.disable_compile.html
|
|
|
|
mlx.core.distributed.all_gather.html
|
|
|
|
mlx.core.distributed.all_sum.html
|
|
|
|
mlx.core.distributed.Group.html
|
|
|
|
mlx.core.distributed.init.html
|
|
|
|
mlx.core.distributed.is_available.html
|
|
|
|
mlx.core.distributed.recv_like.html
|
|
|
|
mlx.core.distributed.recv.html
|
|
|
|
mlx.core.distributed.send.html
|
|
|
|
mlx.core.divide.html
|
|
|
|
mlx.core.divmod.html
|
|
|
|
mlx.core.Dtype.html
|
|
|
|
mlx.core.DtypeCategory.html
|
|
|
|
mlx.core.einsum_path.html
|
|
|
|
mlx.core.einsum.html
|
|
|
|
mlx.core.enable_compile.html
|
|
|
|
mlx.core.equal.html
|
|
|
|
mlx.core.erf.html
|
|
|
|
mlx.core.erfinv.html
|
|
|
|
mlx.core.eval.html
|
|
|
|
mlx.core.exp.html
|
|
|
|
mlx.core.expand_dims.html
|
|
|
|
mlx.core.expm1.html
|
|
|
|
mlx.core.export_function.html
|
|
|
|
mlx.core.export_to_dot.html
|
|
|
|
mlx.core.exporter.html
|
|
|
|
mlx.core.eye.html
|
|
|
|
mlx.core.fast.layer_norm.html
|
|
|
|
mlx.core.fast.metal_kernel.html
|
|
|
|
mlx.core.fast.rms_norm.html
|
|
|
|
mlx.core.fast.rope.html
|
|
|
|
mlx.core.fast.scaled_dot_product_attention.html
|
|
|
|
mlx.core.fft.fft2.html
|
|
|
|
mlx.core.fft.fft.html
|
|
|
|
mlx.core.fft.fftn.html
|
|
|
|
mlx.core.fft.fftshift.html
|
|
|
|
mlx.core.fft.ifft2.html
|
|
|
|
mlx.core.fft.ifft.html
|
|
|
|
mlx.core.fft.ifftn.html
|
|
|
|
mlx.core.fft.ifftshift.html
|
|
|
|
mlx.core.fft.irfft2.html
|
|
|
|
mlx.core.fft.irfft.html
|
|
|
|
mlx.core.fft.irfftn.html
|
|
|
|
mlx.core.fft.rfft2.html
|
|
|
|
mlx.core.fft.rfft.html
|
|
|
|
mlx.core.fft.rfftn.html
|
|
|
|
mlx.core.finfo.html
|
|
|
|
mlx.core.flatten.html
|
|
|
|
mlx.core.floor_divide.html
|
|
|
|
mlx.core.floor.html
|
|
|
|
mlx.core.full.html
|
|
|
|
mlx.core.gather_mm.html
|
|
|
|
mlx.core.gather_qmm.html
|
|
|
|
mlx.core.get_active_memory.html
|
|
|
|
mlx.core.get_cache_memory.html
|
|
|
|
mlx.core.get_peak_memory.html
|
|
|
|
mlx.core.grad.html
|
|
|
|
mlx.core.greater_equal.html
|
|
|
|
mlx.core.greater.html
|
|
|
|
mlx.core.hadamard_transform.html
|
|
|
|
mlx.core.identity.html
|
|
|
|
mlx.core.imag.html
|
|
|
|
mlx.core.import_function.html
|
|
|
|
mlx.core.inner.html
|
|
|
|
mlx.core.isclose.html
|
|
|
|
mlx.core.isfinite.html
|
|
|
|
mlx.core.isinf.html
|
|
|
|
mlx.core.isnan.html
|
|
|
|
mlx.core.isneginf.html
|
|
|
|
mlx.core.isposinf.html
|
|
|
|
mlx.core.issubdtype.html
|
|
|
|
mlx.core.jvp.html
|
|
|
|
mlx.core.kron.html
|
|
|
|
mlx.core.left_shift.html
|
|
|
|
mlx.core.less_equal.html
|
|
|
|
mlx.core.less.html
|
|
|
|
mlx.core.linalg.cholesky_inv.html
|
|
|
|
mlx.core.linalg.cholesky.html
|
|
|
|
mlx.core.linalg.cross.html
|
|
|
|
mlx.core.linalg.eig.html
|
|
|
|
mlx.core.linalg.eigh.html
|
|
|
|
mlx.core.linalg.eigvals.html
|
|
|
|
mlx.core.linalg.eigvalsh.html
|
|
|
|
mlx.core.linalg.inv.html
|
|
|
|
mlx.core.linalg.lu_factor.html
|
|
|
|
mlx.core.linalg.lu.html
|
|
|
|
mlx.core.linalg.norm.html
|
|
|
|
mlx.core.linalg.pinv.html
|
|
|
|
mlx.core.linalg.qr.html
|
|
|
|
mlx.core.linalg.solve_triangular.html
|
|
|
|
mlx.core.linalg.solve.html
|
|
|
|
mlx.core.linalg.svd.html
|
|
|
|
mlx.core.linalg.tri_inv.html
|
|
|
|
mlx.core.linspace.html
|
|
|
|
mlx.core.load.html
|
|
|
|
mlx.core.log1p.html
|
|
|
|
mlx.core.log2.html
|
|
|
|
mlx.core.log10.html
|
|
|
|
mlx.core.log.html
|
|
|
|
mlx.core.logaddexp.html
|
|
|
|
mlx.core.logcumsumexp.html
|
|
|
|
mlx.core.logical_and.html
|
|
|
|
mlx.core.logical_not.html
|
|
|
|
mlx.core.logical_or.html
|
|
|
|
mlx.core.logsumexp.html
|
|
|
|
mlx.core.matmul.html
|
|
|
|
mlx.core.max.html
|
|
|
|
mlx.core.maximum.html
|
|
|
|
mlx.core.mean.html
|
|
|
|
mlx.core.meshgrid.html
|
|
|
|
mlx.core.metal.device_info.html
|
|
|
|
mlx.core.metal.is_available.html
|
|
|
|
mlx.core.metal.start_capture.html
|
|
|
|
mlx.core.metal.stop_capture.html
|
|
|
|
mlx.core.min.html
|
|
|
|
mlx.core.minimum.html
|
|
|
|
mlx.core.moveaxis.html
|
|
|
|
mlx.core.multiply.html
|
|
|
|
mlx.core.nan_to_num.html
|
|
|
|
mlx.core.negative.html
|
|
|
|
mlx.core.new_stream.html
|
|
|
|
mlx.core.not_equal.html
|
|
|
|
mlx.core.ones_like.html
|
|
|
|
mlx.core.ones.html
|
|
|
|
mlx.core.outer.html
|
|
|
|
mlx.core.pad.html
|
|
|
|
mlx.core.partition.html
|
|
|
|
mlx.core.power.html
|
|
|
|
mlx.core.prod.html
|
|
|
|
mlx.core.put_along_axis.html
|
|
|
|
mlx.core.quantize.html
|
|
|
|
mlx.core.quantized_matmul.html
|
|
|
|
mlx.core.radians.html
|
|
|
|
mlx.core.random.bernoulli.html
|
|
|
|
mlx.core.random.categorical.html
|
|
|
|
mlx.core.random.gumbel.html
|
|
|
|
mlx.core.random.key.html
|
|
|
|
mlx.core.random.laplace.html
|
|
|
|
mlx.core.random.multivariate_normal.html
|
|
|
|
mlx.core.random.normal.html
|
|
|
|
mlx.core.random.permutation.html
|
|
|
|
mlx.core.random.randint.html
|
|
|
|
mlx.core.random.seed.html
|
|
|
|
mlx.core.random.split.html
|
|
|
|
mlx.core.random.truncated_normal.html
|
|
|
|
mlx.core.random.uniform.html
|
|
|
|
mlx.core.real.html
|
|
|
|
mlx.core.reciprocal.html
|
|
|
|
mlx.core.remainder.html
|
|
|
|
mlx.core.repeat.html
|
|
|
|
mlx.core.reset_peak_memory.html
|
|
|
|
mlx.core.reshape.html
|
|
|
|
mlx.core.right_shift.html
|
|
|
|
mlx.core.roll.html
|
|
|
|
mlx.core.round.html
|
|
|
|
mlx.core.rsqrt.html
|
|
|
|
mlx.core.save_gguf.html
|
|
|
|
mlx.core.save_safetensors.html
|
|
|
|
mlx.core.save.html
|
|
|
|
mlx.core.savez_compressed.html
|
|
|
|
mlx.core.savez.html
|
|
|
|
mlx.core.set_cache_limit.html
|
|
|
|
mlx.core.set_default_device.html
|
|
|
|
mlx.core.set_default_stream.html
|
|
|
|
mlx.core.set_memory_limit.html
|
|
|
|
mlx.core.set_wired_limit.html
|
|
|
|
mlx.core.sigmoid.html
|
|
|
|
mlx.core.sign.html
|
|
|
|
mlx.core.sin.html
|
|
|
|
mlx.core.sinh.html
|
|
|
|
mlx.core.slice_update.html
|
|
|
|
mlx.core.slice.html
|
|
|
|
mlx.core.softmax.html
|
|
|
|
mlx.core.sort.html
|
|
|
|
mlx.core.split.html
|
|
|
|
mlx.core.sqrt.html
|
|
|
|
mlx.core.square.html
|
|
|
|
mlx.core.squeeze.html
|
|
|
|
mlx.core.stack.html
|
|
|
|
mlx.core.std.html
|
|
|
|
mlx.core.stop_gradient.html
|
|
|
|
mlx.core.stream.html
|
|
|
|
mlx.core.subtract.html
|
|
|
|
mlx.core.sum.html
|
|
|
|
mlx.core.swapaxes.html
|
|
|
|
mlx.core.synchronize.html
|
|
|
|
mlx.core.take_along_axis.html
|
|
|
|
mlx.core.take.html
|
|
|
|
mlx.core.tan.html
|
|
|
|
mlx.core.tanh.html
|
|
|
|
mlx.core.tensordot.html
|
|
|
|
mlx.core.tile.html
|
|
|
|
mlx.core.topk.html
|
|
|
|
mlx.core.trace.html
|
|
|
|
mlx.core.transpose.html
|
|
|
|
mlx.core.tri.html
|
|
|
|
mlx.core.tril.html
|
|
|
|
mlx.core.triu.html
|
|
|
|
mlx.core.unflatten.html
|
|
|
|
mlx.core.value_and_grad.html
|
|
|
|
mlx.core.var.html
|
|
|
|
mlx.core.view.html
|
|
|
|
mlx.core.vjp.html
|
|
|
|
mlx.core.vmap.html
|
|
|
|
mlx.core.where.html
|
|
|
|
mlx.core.zeros_like.html
|
|
|
|
mlx.core.zeros.html
|
|
|
|
mlx.nn.average_gradients.html
|
|
|
|
mlx.nn.quantize.html
|
|
|
|
mlx.nn.value_and_grad.html
|
|
|
|
mlx.optimizers.clip_grad_norm.html
|
|
|
|
mlx.utils.tree_flatten.html
|
|
|
|
mlx.utils.tree_map_with_path.html
|
|
|
|
mlx.utils.tree_map.html
|
|
|
|
mlx.utils.tree_reduce.html
|
|
|
|
mlx.utils.tree_unflatten.html
|
|
|
|
stream_class.html
|
|
|