pytorch

e2361bcc - [functorch] Add exhaustive testing of vmap autograd composability (pytorch/functorch#851)

[functorch] Add exhaustive testing of vmap autograd composability (pytorch/functorch#851) * refactor to make simpler based on comments * cleanup * more failing tests * fix test failures * more test failures * update xfails