Searched refs:f_jvp_traceable (Results 1 – 1 of 1) sorted by relevance
638 f_jvp, out_nonzeros = f_jvp_traceable(jvp(f, instantiate=instantiate), nonzeros)645 def f_jvp_traceable(nonzeros, *primals_and_nztangents): function787 f_jvp, out_nonzeros = f_jvp_traceable(jvp(f, instantiate=instantiate), nonzeros)