mlx.core.array.argmax# array.argmax(self, axis: int | None = None, keepdims: bool = False, *, stream: Stream | ThreadLocalStream | Device | mlx.core.DeviceType | None = None) → array# See argmax().