GPU AI optimization