pytorch

9cba6926 - [functorch] Added VMAP_SUPPORT for returning vectors of tensors