Hi antspy, just to check, do you mean, as long as we keep our state outside fo the function, and pass it in as a parameter, no need for backward?
so, eg I can do something like:
a = autograd.Variable(...)
def f(input, a):
   # stuff here
   return result

out = f(input, a)

… and all should be ok?