Hi,
thank you for your response! Unfortunately, I cannot write the code without the loop (believe me, I’ve tried - I am well aware of the performance hit I take there). From the look of things, I cannot parallelize that loop while maintaining gradients, correct?
Thanks!