pytorch

5b6b1711 - [functorch] Enabled pytree of tensors output for jvp, jacfwd (pytorch/functorch#404)

[functorch] Enabled pytree of tensors output for jvp, jacfwd (pytorch/functorch#404) * Enabled non-tensor output for jvp, jacfwd * Added jacfwd tests * Do not accept functions returning non-tensor output * Updated tests and vjp output checking * Updates according to the review and assert_non_empty_output -> assert_non_empty_tensor_output * Added regexps * More updates