Given a float tensor x, compute the gradient of y = (x ** 2).sum() with respect to x using automatic differentiation, and return the resulting gradient tensor (which equals 2 * x).

Hint: enable gradient tracking on a copy of x, run the forward computation, then trigger backpropagation to populate the gradient.