| .. |
|
mlx.core.abs.doctree
|
|
|
|
mlx.core.add.doctree
|
|
|
|
mlx.core.all.doctree
|
|
|
|
mlx.core.allclose.doctree
|
|
|
|
mlx.core.any.doctree
|
|
|
|
mlx.core.arange.doctree
|
|
|
|
mlx.core.arccos.doctree
|
|
|
|
mlx.core.arccosh.doctree
|
|
|
|
mlx.core.arcsin.doctree
|
|
|
|
mlx.core.arcsinh.doctree
|
|
|
|
mlx.core.arctan.doctree
|
|
|
|
mlx.core.arctanh.doctree
|
|
|
|
mlx.core.argmax.doctree
|
|
|
|
mlx.core.argmin.doctree
|
|
|
|
mlx.core.argpartition.doctree
|
|
|
|
mlx.core.argsort.doctree
|
|
|
|
mlx.core.array_equal.doctree
|
|
|
|
mlx.core.array.abs.doctree
|
|
|
|
mlx.core.array.all.doctree
|
|
|
|
mlx.core.array.any.doctree
|
|
|
|
mlx.core.array.argmax.doctree
|
|
|
|
mlx.core.array.argmin.doctree
|
|
|
|
mlx.core.array.astype.doctree
|
|
|
|
mlx.core.array.cos.doctree
|
|
|
|
mlx.core.array.doctree
|
|
|
|
mlx.core.array.dtype.doctree
|
|
|
|
mlx.core.array.exp.doctree
|
|
|
|
mlx.core.array.item.doctree
|
|
|
|
mlx.core.array.log1p.doctree
|
|
|
|
mlx.core.array.log.doctree
|
|
|
|
mlx.core.array.logsumexp.doctree
|
|
|
|
mlx.core.array.max.doctree
|
|
|
|
mlx.core.array.mean.doctree
|
|
|
|
mlx.core.array.min.doctree
|
|
|
|
mlx.core.array.ndim.doctree
|
|
|
|
mlx.core.array.prod.doctree
|
|
|
|
mlx.core.array.reciprocal.doctree
|
|
|
|
mlx.core.array.reshape.doctree
|
|
|
|
mlx.core.array.round.doctree
|
|
|
|
mlx.core.array.rsqrt.doctree
|
|
|
|
mlx.core.array.shape.doctree
|
|
|
|
mlx.core.array.sin.doctree
|
|
|
|
mlx.core.array.size.doctree
|
|
|
|
mlx.core.array.split.doctree
|
|
|
|
mlx.core.array.sqrt.doctree
|
|
|
|
mlx.core.array.square.doctree
|
|
|
|
mlx.core.array.sum.doctree
|
|
|
|
mlx.core.array.T.doctree
|
|
|
|
mlx.core.array.tolist.doctree
|
|
|
|
mlx.core.array.transpose.doctree
|
|
|
|
mlx.core.array.var.doctree
|
|
|
|
mlx.core.broadcast_to.doctree
|
|
|
|
mlx.core.ceil.doctree
|
|
|
|
mlx.core.clip.doctree
|
|
|
|
mlx.core.concatenate.doctree
|
|
|
|
mlx.core.conv1d.doctree
|
|
|
|
mlx.core.conv2d.doctree
|
|
|
|
mlx.core.convolve.doctree
|
|
|
|
mlx.core.cos.doctree
|
|
|
|
mlx.core.cosh.doctree
|
|
|
|
mlx.core.default_device.doctree
|
|
|
|
mlx.core.default_stream.doctree
|
|
|
|
mlx.core.dequantize.doctree
|
|
|
|
mlx.core.Device.doctree
|
|
|
|
mlx.core.divide.doctree
|
|
|
|
mlx.core.divmod.doctree
|
|
|
|
mlx.core.Dtype.doctree
|
|
|
|
mlx.core.equal.doctree
|
|
|
|
mlx.core.erf.doctree
|
|
|
|
mlx.core.erfinv.doctree
|
|
|
|
mlx.core.eval.doctree
|
|
|
|
mlx.core.exp.doctree
|
|
|
|
mlx.core.expand_dims.doctree
|
|
|
|
mlx.core.eye.doctree
|
|
|
|
mlx.core.fft.fft2.doctree
|
|
|
|
mlx.core.fft.fft.doctree
|
|
|
|
mlx.core.fft.fftn.doctree
|
|
|
|
mlx.core.fft.ifft2.doctree
|
|
|
|
mlx.core.fft.ifft.doctree
|
|
|
|
mlx.core.fft.ifftn.doctree
|
|
|
|
mlx.core.fft.irfft2.doctree
|
|
|
|
mlx.core.fft.irfft.doctree
|
|
|
|
mlx.core.fft.irfftn.doctree
|
|
|
|
mlx.core.fft.rfft2.doctree
|
|
|
|
mlx.core.fft.rfft.doctree
|
|
|
|
mlx.core.fft.rfftn.doctree
|
|
|
|
mlx.core.flatten.doctree
|
|
|
|
mlx.core.floor_divide.doctree
|
|
|
|
mlx.core.floor.doctree
|
|
|
|
mlx.core.full.doctree
|
|
|
|
mlx.core.grad.doctree
|
|
|
|
mlx.core.greater_equal.doctree
|
|
|
|
mlx.core.greater.doctree
|
|
|
|
mlx.core.identity.doctree
|
|
|
|
mlx.core.inner.doctree
|
|
|
|
mlx.core.isinf.doctree
|
|
|
|
mlx.core.isnan.doctree
|
|
|
|
mlx.core.isneginf.doctree
|
|
|
|
mlx.core.isposinf.doctree
|
|
|
|
mlx.core.jvp.doctree
|
|
|
|
mlx.core.less_equal.doctree
|
|
|
|
mlx.core.less.doctree
|
|
|
|
mlx.core.linalg.norm.doctree
|
|
|
|
mlx.core.linspace.doctree
|
|
|
|
mlx.core.load.doctree
|
|
|
|
mlx.core.log1p.doctree
|
|
|
|
mlx.core.log2.doctree
|
|
|
|
mlx.core.log10.doctree
|
|
|
|
mlx.core.log.doctree
|
|
|
|
mlx.core.logaddexp.doctree
|
|
|
|
mlx.core.logical_and.doctree
|
|
|
|
mlx.core.logical_not.doctree
|
|
|
|
mlx.core.logical_or.doctree
|
|
|
|
mlx.core.logsumexp.doctree
|
|
|
|
mlx.core.matmul.doctree
|
|
|
|
mlx.core.max.doctree
|
|
|
|
mlx.core.maximum.doctree
|
|
|
|
mlx.core.mean.doctree
|
|
|
|
mlx.core.min.doctree
|
|
|
|
mlx.core.minimum.doctree
|
|
|
|
mlx.core.moveaxis.doctree
|
|
|
|
mlx.core.multiply.doctree
|
|
|
|
mlx.core.negative.doctree
|
|
|
|
mlx.core.new_stream.doctree
|
|
|
|
mlx.core.ones_like.doctree
|
|
|
|
mlx.core.ones.doctree
|
|
|
|
mlx.core.outer.doctree
|
|
|
|
mlx.core.pad.doctree
|
|
|
|
mlx.core.partition.doctree
|
|
|
|
mlx.core.prod.doctree
|
|
|
|
mlx.core.quantize.doctree
|
|
|
|
mlx.core.quantized_matmul.doctree
|
|
|
|
mlx.core.random.bernoulli.doctree
|
|
|
|
mlx.core.random.categorical.doctree
|
|
|
|
mlx.core.random.gumbel.doctree
|
|
|
|
mlx.core.random.key.doctree
|
|
|
|
mlx.core.random.normal.doctree
|
|
|
|
mlx.core.random.randint.doctree
|
|
|
|
mlx.core.random.seed.doctree
|
|
|
|
mlx.core.random.split.doctree
|
|
|
|
mlx.core.random.truncated_normal.doctree
|
|
|
|
mlx.core.random.uniform.doctree
|
|
|
|
mlx.core.reciprocal.doctree
|
|
|
|
mlx.core.repeat.doctree
|
|
|
|
mlx.core.reshape.doctree
|
|
|
|
mlx.core.round.doctree
|
|
|
|
mlx.core.rsqrt.doctree
|
|
|
|
mlx.core.save_gguf.doctree
|
|
|
|
mlx.core.save_safetensors.doctree
|
|
|
|
mlx.core.save.doctree
|
|
|
|
mlx.core.savez_compressed.doctree
|
|
|
|
mlx.core.savez.doctree
|
|
|
|
mlx.core.set_default_device.doctree
|
|
|
|
mlx.core.set_default_stream.doctree
|
|
|
|
mlx.core.sigmoid.doctree
|
|
|
|
mlx.core.sign.doctree
|
|
|
|
mlx.core.simplify.doctree
|
|
|
|
mlx.core.sin.doctree
|
|
|
|
mlx.core.sinh.doctree
|
|
|
|
mlx.core.softmax.doctree
|
|
|
|
mlx.core.sort.doctree
|
|
|
|
mlx.core.split.doctree
|
|
|
|
mlx.core.sqrt.doctree
|
|
|
|
mlx.core.square.doctree
|
|
|
|
mlx.core.squeeze.doctree
|
|
|
|
mlx.core.stack.doctree
|
|
|
|
mlx.core.stop_gradient.doctree
|
|
|
|
mlx.core.Stream.doctree
|
|
|
|
mlx.core.subtract.doctree
|
|
|
|
mlx.core.sum.doctree
|
|
|
|
mlx.core.swapaxes.doctree
|
|
|
|
mlx.core.take_along_axis.doctree
|
|
|
|
mlx.core.take.doctree
|
|
|
|
mlx.core.tan.doctree
|
|
|
|
mlx.core.tanh.doctree
|
|
|
|
mlx.core.tensordot.doctree
|
|
|
|
mlx.core.transpose.doctree
|
|
|
|
mlx.core.tri.doctree
|
|
|
|
mlx.core.tril.doctree
|
|
|
|
mlx.core.triu.doctree
|
|
|
|
mlx.core.value_and_grad.doctree
|
|
|
|
mlx.core.var.doctree
|
|
|
|
mlx.core.vjp.doctree
|
|
|
|
mlx.core.vmap.doctree
|
|
|
|
mlx.core.where.doctree
|
|
|
|
mlx.core.zeros_like.doctree
|
|
|
|
mlx.core.zeros.doctree
|
|
|
|
mlx.nn.value_and_grad.doctree
|
|
|
|
mlx.optimizers.AdaDelta.doctree
|
|
|
|
mlx.optimizers.Adagrad.doctree
|
|
|
|
mlx.optimizers.Adam.doctree
|
|
|
|
mlx.optimizers.Adamax.doctree
|
|
|
|
mlx.optimizers.AdamW.doctree
|
|
|
|
mlx.optimizers.Lion.doctree
|
|
|
|
mlx.optimizers.Optimizer.doctree
|
|
|
|
mlx.optimizers.OptimizerState.doctree
|
|
|
|
mlx.optimizers.RMSprop.doctree
|
|
|
|
mlx.optimizers.SGD.doctree
|
|
|
|
mlx.utils.tree_flatten.doctree
|
|
|
|
mlx.utils.tree_map.doctree
|
|
|
|
mlx.utils.tree_unflatten.doctree
|
|
|
