pytorch

1af5d3b0 - [functorch] add new vmap examples, cleanup existing vmap docstring