.. _ops: Operations ========== .. currentmodule:: mlx.core .. autosummary:: :toctree: _autosummary abs add all allclose any arange arccos arccosh arcsin arcsinh arctan arctanh argmax argmin argpartition argsort array_equal broadcast_to ceil concatenate convolve conv1d conv2d cos cosh divide equal erf erfinv exp expand_dims eye floor flatten full greater greater_equal identity less less_equal load log log2 log10 log1p logaddexp logical_not logsumexp matmul max maximum mean min minimum moveaxis multiply negative ones ones_like partition pad prod reciprocal reshape rsqrt save savez savez_compressed sigmoid sign sin sinh softmax sort split sqrt square squeeze stack stop_gradient subtract sum swapaxes take take_along_axis tan tanh transpose tri tril triu var where zeros zeros_like