I’m afraid you cannot modify the graph like that at the moment.
You have to redo the forward.
@smth is that a feature we would like to add in the future? Not sure if it’s even feasible with the current backend.