Consider adding `ArgMax` symbol

[arcondello]

Equivalent to NumPy's argmax() function.

[fastbodin]

I believe I can pilfer things from #364 to implement this pretty quickly.

[arcondello]

We specifically also need it to accept an axis= argument. Not too much worse by 🤷