Thanks a lot! Yes, I’m interested in the primitives (that are not computed using autograd).
Do you know whether I can find python wrappers to THNN and THCUNN so that I can call updateGradInput and AccGradParameters directly from python?