pytorch

fbc76eb6 - [functorch] Add mode dispatch stack (pytorch/functorch#34)

[functorch] Add mode dispatch stack (pytorch/functorch#34) * Add mode dispatch stack * remove some things * fix some type errors and clean up vmap * Added a warning