from here, i know that we can get the argmax of a `vector like this:
argmax
vector v{ 3, 1, -14, 1, 5