I’m about confused here, what is the error and is W1 trained in addition to W0? It seems a bit strange to say W1 is being trained if it doesn’t appear to be a leaf node of the computation graph. Additionally, how would W1’s be computed if func_1 as described isn’t differentiable?