pytorch

a598ed6b - [functorch] Added softmax batching rule (tested through local OpInfo, to be upstreamed)