I want to find the indices of the maximum values along axis 0 in a multi dimensional array. This can be done in `python` using `numpy`: <https://docs.scipy.org/doc/numpy/reference/generated/numpy.argmax.html>

In `python`:

```
a = array([[1,2,3],[4,3,1]])
a.argmax(axis=0)

```

Will return:

```
array([1, 1, 0])

```

How would one do the exact same thing in `swift`?