| .. |
|
mlx.core.abs.html
|
|
|
|
mlx.core.add.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.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.cos.html
|
|
|
|
mlx.core.array.dtype.html
|
|
|
|
mlx.core.array.exp.html
|
|
|
|
mlx.core.array.html
|
|
|
|
mlx.core.array.item.html
|
|
|
|
mlx.core.array.log1p.html
|
|
|
|
mlx.core.array.log.html
|
|
|
|
mlx.core.array.logsumexp.html
|
|
|
|
mlx.core.array.max.html
|
|
|
|
mlx.core.array.mean.html
|
|
|
|
mlx.core.array.min.html
|
|
|
|
mlx.core.array.ndim.html
|
|
|
|
mlx.core.array.prod.html
|
|
|
|
mlx.core.array.reciprocal.html
|
|
|
|
mlx.core.array.reshape.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.sum.html
|
|
|
|
mlx.core.array.T.html
|
|
|
|
mlx.core.array.tolist.html
|
|
|
|
mlx.core.array.transpose.html
|
|
|
|
mlx.core.array.var.html
|
|
|
|
mlx.core.broadcast_to.html
|
|
|
|
mlx.core.concatenate.html
|
|
|
|
mlx.core.conv1d.html
|
|
|
|
mlx.core.conv2d.html
|
|
|
|
mlx.core.convolve.html
|
|
|
|
mlx.core.cos.html
|
|
|
|
mlx.core.cosh.html
|
|
|
|
mlx.core.default_device.html
|
|
|
|
mlx.core.default_stream.html
|
|
|
|
mlx.core.Device.html
|
|
|
|
mlx.core.divide.html
|
|
|
|
mlx.core.Dtype.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.fft.fft2.html
|
|
|
|
mlx.core.fft.fft.html
|
|
|
|
mlx.core.fft.fftn.html
|
|
|
|
mlx.core.fft.ifft2.html
|
|
|
|
mlx.core.fft.ifft.html
|
|
|
|
mlx.core.fft.ifftn.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.full.html
|
|
|
|
mlx.core.grad.html
|
|
|
|
mlx.core.greater_equal.html
|
|
|
|
mlx.core.greater.html
|
|
|
|
mlx.core.jvp.html
|
|
|
|
mlx.core.less_equal.html
|
|
|
|
mlx.core.less.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.logical_not.html
|
|
|
|
mlx.core.logsumexp.html
|
|
|
|
mlx.core.matmul.html
|
|
|
|
mlx.core.max.html
|
|
|
|
mlx.core.maximum.html
|
|
|
|
mlx.core.mean.html
|
|
|
|
mlx.core.min.html
|
|
|
|
mlx.core.minimum.html
|
|
|
|
mlx.core.multiply.html
|
|
|
|
mlx.core.negative.html
|
|
|
|
mlx.core.new_stream.html
|
|
|
|
mlx.core.ones_like.html
|
|
|
|
mlx.core.ones.html
|
|
|
|
mlx.core.pad.html
|
|
|
|
mlx.core.partition.html
|
|
|
|
mlx.core.prod.html
|
|
|
|
mlx.core.random.bernoulli.html
|
|
|
|
mlx.core.random.categorical.html
|
|
|
|
mlx.core.random.gumbel.html
|
|
|
|
mlx.core.random.key.html
|
|
|
|
mlx.core.random.normal.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.reciprocal.html
|
|
|
|
mlx.core.reshape.html
|
|
|
|
mlx.core.rsqrt.html
|
|
|
|
mlx.core.save.html
|
|
|
|
mlx.core.savez_compressed.html
|
|
|
|
mlx.core.savez.html
|
|
|
|
mlx.core.set_default_device.html
|
|
|
|
mlx.core.set_default_stream.html
|
|
|
|
mlx.core.sigmoid.html
|
|
|
|
mlx.core.sign.html
|
|
|
|
mlx.core.sin.html
|
|
|
|
mlx.core.sinh.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.stop_gradient.html
|
|
|
|
mlx.core.Stream.html
|
|
|
|
mlx.core.subtract.html
|
|
|
|
mlx.core.sum.html
|
|
|
|
mlx.core.take_along_axis.html
|
|
|
|
mlx.core.take.html
|
|
|
|
mlx.core.tan.html
|
|
|
|
mlx.core.tanh.html
|
|
|
|
mlx.core.transpose.html
|
|
|
|
mlx.core.value_and_grad.html
|
|
|
|
mlx.core.var.html
|
|
|
|
mlx.core.vjp.html
|
|
|
|
mlx.core.vmap.html
|
|
|
|
mlx.core.where.html
|
|
|
|
mlx.core.zeros_like.html
|
|
|
|
mlx.core.zeros.html
|
|
|
|
mlx.nn.Conv1d.html
|
|
|
|
mlx.nn.Conv2d.html
|
|
|
|
mlx.nn.Embedding.html
|
|
|
|
mlx.nn.GELU.html
|
|
|
|
mlx.nn.GroupNorm.html
|
|
|
|
mlx.nn.LayerNorm.html
|
|
|
|
mlx.nn.Linear.html
|
|
|
|
mlx.nn.MultiHeadAttention.html
|
|
|
|
mlx.nn.ReLU.html
|
|
|
|
mlx.nn.RMSNorm.html
|
|
|
|
mlx.nn.RoPE.html
|
|
|
|
mlx.nn.Sequential.html
|
|
|
|
mlx.nn.SiLU.html
|
|
|
|
mlx.nn.value_and_grad.html
|
|
|
|
mlx.optimizers.Adam.html
|
|
|
|
mlx.optimizers.Optimizer.html
|
|
|
|
mlx.optimizers.OptimizerState.html
|
|
|
|
mlx.optimizers.SGD.html
|
|
|
|
mlx.utils.tree_flatten.html
|
|
|
|
mlx.utils.tree_map.html
|
|
|
|
mlx.utils.tree_unflatten.html
|
|
|