mlx.core.array.argmax

array.argmax(self, axis: int | None = None, keepdims: bool = False, *, stream: Stream | Device | None = None) → array

See argmax().