pytorch

afba1985 - [functorch] some minor fixes