pytorch

bf1df4f3 - [functorch] ported sumdim batch rule over and added argmax