Searched defs:vjpfun (Results 1 – 3 of 3) sorted by relevance
744 vjpfun = lambda ct: [vjp(ct, *primals) if vjp else zeros_like_jaxval(x) function752 vjpfun = lambda ct: [vjp(ct, ans, *primals) if vjp else zeros_like_jaxval(x) function
85 def translate_vjp(vjpfun, fun, argnum): argument
2637 def vjpfun(ct): function