pytorch

e19546fb - [functorch] jvp transform accepts functions that take multiple inputs and outputs (pytorch/functorch#267)