| .. |
|
mlx.core.abs.rst
|
|
|
|
mlx.core.add.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.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.cos.rst
|
|
|
|
mlx.core.array.dtype.rst
|
|
|
|
mlx.core.array.exp.rst
|
|
|
|
mlx.core.array.item.rst
|
|
|
|
mlx.core.array.log1p.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.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.sum.rst
|
|
|
|
mlx.core.array.T.rst
|
|
|
|
mlx.core.array.tolist.rst
|
|
|
|
mlx.core.array.transpose.rst
|
|
|
|
mlx.core.array.var.rst
|
|
|
|
mlx.core.broadcast_to.rst
|
|
|
|
mlx.core.ceil.rst
|
|
|
|
mlx.core.clip.rst
|
|
|
|
mlx.core.concatenate.rst
|
|
|
|
mlx.core.conv1d.rst
|
|
|
|
mlx.core.conv2d.rst
|
|
|
|
mlx.core.convolve.rst
|
|
|
|
mlx.core.cos.rst
|
|
|
|
mlx.core.cosh.rst
|
|
|
|
mlx.core.default_device.rst
|
|
|
|
mlx.core.default_stream.rst
|
|
|
|
mlx.core.dequantize.rst
|
|
|
|
mlx.core.Device.rst
|
|
|
|
mlx.core.divide.rst
|
|
|
|
mlx.core.Dtype.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.eye.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.flatten.rst
|
|
|
|
mlx.core.floor_divide.rst
|
|
|
|
mlx.core.floor.rst
|
|
|
|
mlx.core.full.rst
|
|
|
|
mlx.core.grad.rst
|
|
|
|
mlx.core.greater_equal.rst
|
|
|
|
mlx.core.greater.rst
|
|
|
|
mlx.core.identity.rst
|
|
|
|
mlx.core.jvp.rst
|
|
|
|
mlx.core.less_equal.rst
|
|
|
|
mlx.core.less.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_not.rst
|
|
|
|
mlx.core.logsumexp.rst
|
|
|
|
mlx.core.matmul.rst
|
|
|
|
mlx.core.max.rst
|
|
|
|
mlx.core.maximum.rst
|
|
|
|
mlx.core.mean.rst
|
|
|
|
mlx.core.min.rst
|
|
|
|
mlx.core.minimum.rst
|
|
|
|
mlx.core.moveaxis.rst
|
|
|
|
mlx.core.multiply.rst
|
|
|
|
mlx.core.negative.rst
|
|
|
|
mlx.core.new_stream.rst
|
|
|
|
mlx.core.ones_like.rst
|
|
|
|
mlx.core.ones.rst
|
|
|
|
mlx.core.pad.rst
|
|
|
|
mlx.core.partition.rst
|
|
|
|
mlx.core.prod.rst
|
|
|
|
mlx.core.quantize.rst
|
|
|
|
mlx.core.quantized_matmul.rst
|
|
|
|
mlx.core.random.bernoulli.rst
|
|
|
|
mlx.core.random.categorical.rst
|
|
|
|
mlx.core.random.gumbel.rst
|
|
|
|
mlx.core.random.key.rst
|
|
|
|
mlx.core.random.normal.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.reciprocal.rst
|
|
|
|
mlx.core.reshape.rst
|
|
|
|
mlx.core.round.rst
|
|
|
|
mlx.core.rsqrt.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.simplify.rst
|
|
|
|
mlx.core.sin.rst
|
|
|
|
mlx.core.sinh.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.stop_gradient.rst
|
|
|
|
mlx.core.Stream.rst
|
|
|
|
mlx.core.subtract.rst
|
|
|
|
mlx.core.sum.rst
|
|
|
|
mlx.core.swapaxes.rst
|
|
|
|
mlx.core.take_along_axis.rst
|
|
|
|
mlx.core.take.rst
|
|
|
|
mlx.core.tan.rst
|
|
|
|
mlx.core.tanh.rst
|
|
|
|
mlx.core.transpose.rst
|
|
|
|
mlx.core.tri.rst
|
|
|
|
mlx.core.tril.rst
|
|
|
|
mlx.core.triu.rst
|
|
|
|
mlx.core.value_and_grad.rst
|
|
|
|
mlx.core.var.rst
|
|
|
|
mlx.core.vjp.rst
|
|
|
|
mlx.core.vmap.rst
|
|
|
|
mlx.core.where.rst
|
|
|
|
mlx.core.zeros_like.rst
|
|
|
|
mlx.core.zeros.rst
|
|
|
|
mlx.nn.Module.rst
|
|
|
|
mlx.nn.value_and_grad.rst
|
|
|
|
mlx.optimizers.AdaDelta.rst
|
|
|
|
mlx.optimizers.Adagrad.rst
|
|
|
|
mlx.optimizers.Adam.rst
|
|
|
|
mlx.optimizers.Adamax.rst
|
|
|
|
mlx.optimizers.AdamW.rst
|
|
|
|
mlx.optimizers.Lion.rst
|
|
|
|
mlx.optimizers.Optimizer.rst
|
|
|
|
mlx.optimizers.OptimizerState.rst
|
|
|
|
mlx.optimizers.RMSprop.rst
|
|
|
|
mlx.optimizers.SGD.rst
|
|
|
|
mlx.utils.tree_flatten.rst
|
|
|
|
mlx.utils.tree_map.rst
|
|
|
|
mlx.utils.tree_unflatten.rst
|
|
|
