pytorch

2e774b12 - [functorch] Get rid of new_blah_hacks; diagflat batch rule