pytorch

ed2f5824 - [functorch] fix some bugs with reduction batching rule + add amax/amin