.. _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_invert bitwise_or bitwise_xor block_masked_mm broadcast_arrays broadcast_to ceil clip concatenate contiguous conj conjugate convolve conv1d conv2d conv3d conv_transpose1d conv_transpose2d conv_transpose3d conv_general cos cosh cummax cummin cumprod cumsum degrees dequantize diag diagonal divide divmod einsum einsum_path equal erf erfinv exp expm1 expand_dims eye flatten floor floor_divide full gather_mm gather_qmm greater greater_equal hadamard_transform identity imag inner isfinite isclose isinf isnan isneginf isposinf issubdtype kron left_shift less less_equal linspace load log log2 log10 log1p logaddexp logcumsumexp logical_not logical_and logical_or logsumexp matmul max maximum mean meshgrid min minimum moveaxis multiply nan_to_num negative not_equal ones ones_like outer partition pad power prod put_along_axis quantize quantized_matmul radians real reciprocal remainder repeat reshape right_shift roll round rsqrt save savez savez_compressed save_gguf save_safetensors sigmoid sign sin sinh slice slice_update 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 unflatten var view where zeros zeros_like