max and argmax code example

Example: argmax vs max

Argmax is an operation that finds the argument that gives the maximum value from a target function