Perhaps you would like to add some text and/or maths explaining to the unwitting reader why this is nabla_grad rather than grad.