Hi Simon! Thanks for your suggestion. Can you give a short example of how to use nn.functional interface so as to change the parameters w.r.t which the gradients will be computed during the backward pass?