pytorch

1eb454f5 - [functorch] Fix vmapvjp tests (pytorch/functorch#354)

[functorch] Fix vmapvjp tests (pytorch/functorch#354) Previously they were not vmapping over non-differentiable arguments. That case is important to test. Test Plan: - run tests