.. _ops: Operations ========== .. currentmodule:: mlx.core .. autosummary:: :toctree: _autosummary abs add addmm all allclose any arange arccos arccosh arcsin arcsinh arctan arctan2 arctanh argmax argmin argpartition argsort array_equal as_strided atleast_1d atleast_2d atleast_3d bitwise_and bitwise_or bitwise_xor block_masked_mm broadcast_to ceil clip concatenate conj conjugate convolve conv1d conv2d conv_general cos cosh cummax cummin cumprod cumsum degrees dequantize diag diagonal divide divmod equal erf erfinv exp expm1 expand_dims eye flatten floor floor_divide full gather_mm gather_qmm greater greater_equal identity inner isclose isinf isnan isneginf isposinf issubdtype left_shift less less_equal linspace load log log2 log10 log1p logaddexp logical_not logical_and logical_or logsumexp matmul max maximum mean meshgrid min minimum moveaxis multiply negative not_equal ones ones_like outer partition pad power prod quantize quantized_matmul radians reciprocal remainder repeat reshape right_shift round rsqrt save savez savez_compressed save_gguf save_safetensors sigmoid sign sin sinh softmax sort split sqrt square squeeze stack std stop_gradient subtract sum swapaxes take take_along_axis tan tanh tensordot tile topk trace transpose tri tril triu var view where zeros zeros_like