\usepackage{amsmath}
\DeclareMathOperator*{\argmax}{argmax}