Speed up exact gradients computation
momchilmm opened this issue · 0 comments
momchilmm commented
The backprop through the guided mode computation is currently a bottleneck in the gradient computation. There are some things that could be tried to speed things up, including vectorizing the computation w.r.t. g.