Hi,
Do you see a second backtrace that points to a forward function as mentioned in the message? Can you share it here?
You do a few inplace I think: the optimizer steps are doing inplace ops, also all the ops finishing with _ are inplace (like round_() that you use).