Hi, thanks for your response! In general, it should the output order in backward follow the order of inputs? If in forward I had forward(ctx, alpha, x) then backward should return grad_wrt_alpha, grad_x in that order?