| .. |
|
mlx.core.abs.rst.txt
|
|
|
|
mlx.core.add.rst.txt
|
|
|
|
mlx.core.all.rst.txt
|
|
|
|
mlx.core.allclose.rst.txt
|
|
|
|
mlx.core.any.rst.txt
|
|
|
|
mlx.core.arange.rst.txt
|
|
|
|
mlx.core.arccos.rst.txt
|
|
|
|
mlx.core.arccosh.rst.txt
|
|
|
|
mlx.core.arcsin.rst.txt
|
|
|
|
mlx.core.arcsinh.rst.txt
|
|
|
|
mlx.core.arctan.rst.txt
|
|
|
|
mlx.core.arctanh.rst.txt
|
|
|
|
mlx.core.argmax.rst.txt
|
|
|
|
mlx.core.argmin.rst.txt
|
|
|
|
mlx.core.argpartition.rst.txt
|
|
|
|
mlx.core.argsort.rst.txt
|
|
|
|
mlx.core.array_equal.rst.txt
|
|
|
|
mlx.core.array.abs.rst.txt
|
|
|
|
mlx.core.array.all.rst.txt
|
|
|
|
mlx.core.array.any.rst.txt
|
|
|
|
mlx.core.array.argmax.rst.txt
|
|
|
|
mlx.core.array.argmin.rst.txt
|
|
|
|
mlx.core.array.astype.rst.txt
|
|
|
|
mlx.core.array.cos.rst.txt
|
|
|
|
mlx.core.array.dtype.rst.txt
|
|
|
|
mlx.core.array.exp.rst.txt
|
|
|
|
mlx.core.array.item.rst.txt
|
|
|
|
mlx.core.array.log1p.rst.txt
|
|
|
|
mlx.core.array.log.rst.txt
|
|
|
|
mlx.core.array.logsumexp.rst.txt
|
|
|
|
mlx.core.array.max.rst.txt
|
|
|
|
mlx.core.array.mean.rst.txt
|
|
|
|
mlx.core.array.min.rst.txt
|
|
|
|
mlx.core.array.ndim.rst.txt
|
|
|
|
mlx.core.array.prod.rst.txt
|
|
|
|
mlx.core.array.reciprocal.rst.txt
|
|
|
|
mlx.core.array.reshape.rst.txt
|
|
|
|
mlx.core.array.rsqrt.rst.txt
|
|
|
|
mlx.core.array.rst.txt
|
|
|
|
mlx.core.array.shape.rst.txt
|
|
|
|
mlx.core.array.sin.rst.txt
|
|
|
|
mlx.core.array.size.rst.txt
|
|
|
|
mlx.core.array.split.rst.txt
|
|
|
|
mlx.core.array.sqrt.rst.txt
|
|
|
|
mlx.core.array.square.rst.txt
|
|
|
|
mlx.core.array.sum.rst.txt
|
|
|
|
mlx.core.array.T.rst.txt
|
|
|
|
mlx.core.array.tolist.rst.txt
|
|
|
|
mlx.core.array.transpose.rst.txt
|
|
|
|
mlx.core.array.var.rst.txt
|
|
|
|
mlx.core.broadcast_to.rst.txt
|
|
|
|
mlx.core.concatenate.rst.txt
|
|
|
|
mlx.core.conv1d.rst.txt
|
|
|
|
mlx.core.conv2d.rst.txt
|
|
|
|
mlx.core.convolve.rst.txt
|
|
|
|
mlx.core.cos.rst.txt
|
|
|
|
mlx.core.cosh.rst.txt
|
|
|
|
mlx.core.default_device.rst.txt
|
|
|
|
mlx.core.default_stream.rst.txt
|
|
|
|
mlx.core.Device.rst.txt
|
|
|
|
mlx.core.divide.rst.txt
|
|
|
|
mlx.core.Dtype.rst.txt
|
|
|
|
mlx.core.equal.rst.txt
|
|
|
|
mlx.core.erf.rst.txt
|
|
|
|
mlx.core.erfinv.rst.txt
|
|
|
|
mlx.core.eval.rst.txt
|
|
|
|
mlx.core.exp.rst.txt
|
|
|
|
mlx.core.expand_dims.rst.txt
|
|
|
|
mlx.core.fft.fft2.rst.txt
|
|
|
|
mlx.core.fft.fft.rst.txt
|
|
|
|
mlx.core.fft.fftn.rst.txt
|
|
|
|
mlx.core.fft.ifft2.rst.txt
|
|
|
|
mlx.core.fft.ifft.rst.txt
|
|
|
|
mlx.core.fft.ifftn.rst.txt
|
|
|
|
mlx.core.fft.irfft2.rst.txt
|
|
|
|
mlx.core.fft.irfft.rst.txt
|
|
|
|
mlx.core.fft.irfftn.rst.txt
|
|
|
|
mlx.core.fft.rfft2.rst.txt
|
|
|
|
mlx.core.fft.rfft.rst.txt
|
|
|
|
mlx.core.fft.rfftn.rst.txt
|
|
|
|
mlx.core.full.rst.txt
|
|
|
|
mlx.core.grad.rst.txt
|
|
|
|
mlx.core.greater_equal.rst.txt
|
|
|
|
mlx.core.greater.rst.txt
|
|
|
|
mlx.core.jvp.rst.txt
|
|
|
|
mlx.core.less_equal.rst.txt
|
|
|
|
mlx.core.less.rst.txt
|
|
|
|
mlx.core.load.rst.txt
|
|
|
|
mlx.core.log1p.rst.txt
|
|
|
|
mlx.core.log2.rst.txt
|
|
|
|
mlx.core.log10.rst.txt
|
|
|
|
mlx.core.log.rst.txt
|
|
|
|
mlx.core.logaddexp.rst.txt
|
|
|
|
mlx.core.logical_not.rst.txt
|
|
|
|
mlx.core.logsumexp.rst.txt
|
|
|
|
mlx.core.matmul.rst.txt
|
|
|
|
mlx.core.max.rst.txt
|
|
|
|
mlx.core.maximum.rst.txt
|
|
|
|
mlx.core.mean.rst.txt
|
|
|
|
mlx.core.min.rst.txt
|
|
|
|
mlx.core.minimum.rst.txt
|
|
|
|
mlx.core.multiply.rst.txt
|
|
|
|
mlx.core.negative.rst.txt
|
|
|
|
mlx.core.new_stream.rst.txt
|
|
|
|
mlx.core.ones_like.rst.txt
|
|
|
|
mlx.core.ones.rst.txt
|
|
|
|
mlx.core.pad.rst.txt
|
|
|
|
mlx.core.partition.rst.txt
|
|
|
|
mlx.core.prod.rst.txt
|
|
|
|
mlx.core.random.bernoulli.rst.txt
|
|
|
|
mlx.core.random.categorical.rst.txt
|
|
|
|
mlx.core.random.gumbel.rst.txt
|
|
|
|
mlx.core.random.key.rst.txt
|
|
|
|
mlx.core.random.normal.rst.txt
|
|
|
|
mlx.core.random.randint.rst.txt
|
|
|
|
mlx.core.random.seed.rst.txt
|
|
|
|
mlx.core.random.split.rst.txt
|
|
|
|
mlx.core.random.truncated_normal.rst.txt
|
|
|
|
mlx.core.random.uniform.rst.txt
|
|
|
|
mlx.core.reciprocal.rst.txt
|
|
|
|
mlx.core.reshape.rst.txt
|
|
|
|
mlx.core.rsqrt.rst.txt
|
|
|
|
mlx.core.save.rst.txt
|
|
|
|
mlx.core.savez_compressed.rst.txt
|
|
|
|
mlx.core.savez.rst.txt
|
|
|
|
mlx.core.set_default_device.rst.txt
|
|
|
|
mlx.core.set_default_stream.rst.txt
|
|
|
|
mlx.core.sigmoid.rst.txt
|
|
|
|
mlx.core.sign.rst.txt
|
|
|
|
mlx.core.sin.rst.txt
|
|
|
|
mlx.core.sinh.rst.txt
|
|
|
|
mlx.core.softmax.rst.txt
|
|
|
|
mlx.core.sort.rst.txt
|
|
|
|
mlx.core.split.rst.txt
|
|
|
|
mlx.core.sqrt.rst.txt
|
|
|
|
mlx.core.square.rst.txt
|
|
|
|
mlx.core.squeeze.rst.txt
|
|
|
|
mlx.core.stop_gradient.rst.txt
|
|
|
|
mlx.core.Stream.rst.txt
|
|
|
|
mlx.core.subtract.rst.txt
|
|
|
|
mlx.core.sum.rst.txt
|
|
|
|
mlx.core.take_along_axis.rst.txt
|
|
|
|
mlx.core.take.rst.txt
|
|
|
|
mlx.core.tan.rst.txt
|
|
|
|
mlx.core.tanh.rst.txt
|
|
|
|
mlx.core.transpose.rst.txt
|
|
|
|
mlx.core.value_and_grad.rst.txt
|
|
|
|
mlx.core.var.rst.txt
|
|
|
|
mlx.core.vjp.rst.txt
|
|
|
|
mlx.core.vmap.rst.txt
|
|
|
|
mlx.core.where.rst.txt
|
|
|
|
mlx.core.zeros_like.rst.txt
|
|
|
|
mlx.core.zeros.rst.txt
|
|
|
|
mlx.nn.Conv1d.rst.txt
|
|
|
|
mlx.nn.Conv2d.rst.txt
|
|
|
|
mlx.nn.Embedding.rst.txt
|
|
|
|
mlx.nn.GELU.rst.txt
|
|
|
|
mlx.nn.GroupNorm.rst.txt
|
|
|
|
mlx.nn.LayerNorm.rst.txt
|
|
|
|
mlx.nn.Linear.rst.txt
|
|
|
|
mlx.nn.MultiHeadAttention.rst.txt
|
|
|
|
mlx.nn.ReLU.rst.txt
|
|
|
|
mlx.nn.RMSNorm.rst.txt
|
|
|
|
mlx.nn.RoPE.rst.txt
|
|
|
|
mlx.nn.Sequential.rst.txt
|
|
|
|
mlx.nn.SiLU.rst.txt
|
|
|
|
mlx.nn.value_and_grad.rst.txt
|
|
|
|
mlx.optimizers.Adam.rst.txt
|
|
|
|
mlx.optimizers.Optimizer.rst.txt
|
|
|
|
mlx.optimizers.OptimizerState.rst.txt
|
|
|
|
mlx.optimizers.SGD.rst.txt
|
|
|
|
mlx.utils.tree_flatten.rst.txt
|
|
|
|
mlx.utils.tree_map.rst.txt
|
|
|
|
mlx.utils.tree_unflatten.rst.txt
|
|
|
