pytorch

c91ef2f1 - [functorch] refactored argmax batching rule to support argmin too