Optimize GPU gradient