.. _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 clip concatenate convolve conv1d conv2d cos cosh dequantize divide divmod equal erf erfinv exp expand_dims eye flatten floor floor_divide full greater greater_equal identity inner less less_equal linspace load log log2 log10 log1p logaddexp logical_not logical_and logical_or logsumexp matmul max maximum mean min minimum moveaxis multiply negative ones ones_like outer partition pad prod quantize quantized_matmul reciprocal repeat reshape round rsqrt save savez savez_compressed save_gguf save_safetensors sigmoid sign sin sinh softmax sort split sqrt square squeeze stack stop_gradient subtract sum swapaxes take take_along_axis tan tanh tensordot transpose tri tril triu var where zeros zeros_like